camail-official/discretax
Discretax is a light weight collection of state space models implemented in JAX ⚡️
This project offers a collection of pre-built state space models, which are powerful tools for analyzing sequences of data over time. It takes in sequential data, such as sensor readings or financial time series, and outputs insights and predictions based on the underlying system dynamics. Data scientists and machine learning engineers working with time-series analysis and sequential modeling can use this project.
Available on PyPI.
Use this if you need to analyze and model complex sequential data for forecasting or understanding system behavior.
Not ideal if you are looking for a simple, off-the-shelf regression or classification tool for static datasets.
Stars
32
Forks
5
Language
Python
License
MIT
Category
Last pushed
Mar 09, 2026
Commits (30d)
0
Dependencies
4
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/camail-official/discretax"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable....
google/grain
Library for reading and processing ML training data.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/