#

linax

Here is 1 public repository matching this topic...

camail-official / discretax

Discretax 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 Mar 9, 2026
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."