#

linax

Here is 1 public repository matching this topic...

camail-official / linax

linax is a light weight collection of state space models implemented in JAX ⚡️

machine-learning deep-learning neural-networks ssm equinox state-space-models jax sequence-modeling linax

Updated Oct 28, 2025
Python

Improve this page

Add a description, image, and links to the linax topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the linax topic, visit your repo's landing page and select "manage topics."