PyPI page
Home page
Author:
Cosmo Santoni
License:
Apache-2.0
Summary:
Pure JAX/Flax NNX implementation of the Mamba2 state-space model with state caching, pretrained weight loading, and time-series forecasting.
Latest version:
1.1.2
Required dependencies:
flax
|
jax
|
jaxlib
|
optax
Optional dependencies:
absl-py
|
huggingface_hub
|
numpy
|
pytest
|
safetensors
|
torch
|
transformers
Downloads last day:
8
Downloads last week:
86
Downloads last month:
628