PyPI page
Home page
Author:
Peter Szemraj
Summary:
JAX/Equinox implementation of Megalodon for efficient long-context training and inference
Latest version:
0.2.2
Required dependencies:
equinox
|
jax
|
jaxtyping
|
numpy
|
safetensors
Optional dependencies:
jax
|
pytest
|
ruff
|
torch
Downloads last day:
0
Downloads last week:
5
Downloads last month:
26
7-day smoothing