jaxpole
This is an implementation of a differentiable time-varying all-pole filter in JAX based on torchlpc.
Install
pip install jaxpole
or locally from source
pip install -e '.[dev]'
How to use
import jax.numpy as jnp
import jax
pole = 0.99 * jnp.exp(1j * jnp.pi / 4)
coeffs = jnp.array([-2 * pole.real, pole.real**2 + pole.imag**2])
x = jax.random.normal(jax.random.PRNGKey(0), (1, 1000)) # (B, T)
A = jnp.tile(coeffs, (1, x.shape[-1], 1)) # (B, T, P)
zi = jnp.zeros((1, 2)) # (B, P)
# filter the signal
y = allpole(x, A, zi)
1 maintainer
Metadata
1 maintainer
Release files for jaxpole 0.0.3
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| jaxpole-0.0.3.tar.gz | 9.7 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| jaxpole-0.0.3-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 19.1 kB
Release files / jaxpole-0.0.3.tar.gz
| Download URL | jaxpole-0.0.3.tar.gz |
|---|---|
| Size | 9.7 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
31f4537c0985d51a30c5321afc15e64a03a14a3bca39429f1c42de6ca33c471f
|
|
BLAKE2b-256 checksum How to use checksums |
84c9da59cfc7b09e6904a4ba405c8380e3ae1ccd438099a109341f68ed75aece
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.0 CPython/3.10.13
|
Release files / jaxpole-0.0.3-py3-none-any.whl
| Download URL | jaxpole-0.0.3-py3-none-any.whl |
|---|---|
| Size | 9.4 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
8a25717242b759a8f4c76e87f282ca5f7613165f13fddf949a17c1156416c055
|
|
BLAKE2b-256 checksum How to use checksums |
cf099dc60b82c7ea913a836168b178056d47c4d7317c615bc850b9bb1cb0765a
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.0 CPython/3.10.13
|
