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
|
please-downgrade-to-python-3.13-for-flax
Optional dependencies:
datasets
|
equinox
|
flax
|
pillow
|
pytest
|
tqdm
|
wandb
Downloads last day:
2
Downloads last week:
11
Downloads last month:
43