PyPI page
Home page
Author:
Steven Morad
Summary:
Deep memory and sequence modeling in JAX
Latest version:
0.1.1
Required dependencies:
beartype
|
jax
|
jaxlib
|
jaxtyping
|
optax
Optional dependencies:
datasets
|
equinox
|
flax
|
pillow
|
please-downgrade-to-python-3-13-for-flax
|
pytest
|
tqdm
|
wandb
Downloads last day:
1
Downloads last week:
8
Downloads last month:
33
7-day smoothing