Discretax is a light weight collection of state space models implemented in JAX ⚡️
☆38Mar 9, 2026Updated 7 months ago
Alternatives and similar repositories for discretax
Users that are interested in discretax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- This is the official repository for the pre-print "The Key to State Reduction in Linear Attention: A Rank-based Perspective"☆15Sep 25, 2026Updated last month
- Code for the ICML 2023 paper "Geometric Autoencoders - What You See is What You Decode"☆17Jul 28, 2023Updated 3 years ago
- Oscillatory State-Space Models☆129Mar 1, 2026Updated 7 months ago
- Minimal pretraining script for language modeling in PyTorch. Supporting torch compilation and DDP. It includes a model implementation and…☆52Apr 27, 2026Updated 6 months ago
- Present PDFs from your browser. Including speaker notes, mp4/YouTube/Vimeo support and more! Presentations stay on your machine unless ex…☆41Updated this week
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Code for the paper Depth Memory☆14Oct 14, 2026Updated 2 weeks ago
- Differentiable Hydrodynamical Simulations☆18Updated this week
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆48Sep 10, 2026Updated last month
- Prediction markets for research papers.☆55Updated this week
- Finite Element Exterior Calculus Engine in Rust☆23Oct 6, 2026Updated 3 weeks ago
- Message-Passing Monte Carlo (MPMC)☆27Oct 10, 2024Updated 2 years ago
- A simulation node in your JAX workflow.☆90Oct 23, 2026Updated last week
- Fast Kolmogorov-Arnold Network in JAX, initial experiments☆16May 20, 2024Updated 2 years ago
- VoxEMG circuit and schematics, plus example code for further digital filtering with Bela, etc.☆12Mar 6, 2024Updated 2 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated last year
- ☆36Jul 11, 2017Updated 9 years ago
- A repository for Mixed Precision Training in JAX based on Equinox.☆16Nov 2, 2025Updated 11 months ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆68Jan 23, 2026Updated 9 months ago
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Scattering Transform with Random Paths for Machine Learning☆16Oct 15, 2026Updated 2 weeks ago
- Automatic differentiation for Triton Kernels☆29Aug 12, 2025Updated last year
- A snappy + easy + pretty TUI debugger for Python.☆70May 22, 2026Updated 5 months ago
- ☆12Apr 27, 2025Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Gradient-based Hyperparameter Optimization Over Long Horizons☆14Sep 29, 2021Updated 5 years ago
- The Energy Transformer block, in JAX☆69Dec 14, 2023Updated 2 years ago
- Code for "Log Neural Controlled Differential Equations" (ICML 2024) and "Structured Linear CDEs" (NeurIPS 2025, Spotlight)☆33Jul 24, 2026Updated 3 months ago
- Hydra sweeper integration of our favorite optimization packages, utilizing ask-and-tell interfaces.☆16Nov 14, 2025Updated 11 months ago
- Code of the paper "LLM-to-Flow-Distillation: Teaching Natural Language Goals to Scientific Generators"☆19Oct 14, 2026Updated 2 weeks ago
- Extension of dynamax repository to dynamical systems with continuous-time dynamics and measurements sampled at possibly irregular discret…☆25Updated this week
- Prototype for sparse automatic differentiation in JAX + Julia☆24Jun 17, 2025Updated last year
- Convolutional Differential Operators for Physics-based Deep Learning Study☆26Jun 2, 2026Updated 4 months ago
- Tools for managing BibTeX bibliographies: automatically update preprints to published versions and filter to only cited references.☆120Sep 26, 2026Updated last month
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- NYU FRE-GY 6971 Fixed Income Quant Trading☆11Jan 7, 2019Updated 7 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆43Oct 26, 2020Updated 6 years ago
- Spectral / Galerkin / PINNs experimentation toolkit built on top of JAX for fast differentiable ODE / PDE prototyping, variational forms,…☆21Updated this week
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆66Oct 2, 2026Updated 3 weeks ago
- Causal Modeling with Stationary Diffusions, AISTATS 2024☆23Mar 3, 2025Updated last year
- ProbPy is a comprehensive repository dedicated to providing an extensive collection of probability puzzles, riddles, and solutions typica…☆12Jun 27, 2023Updated 3 years ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆71Aug 1, 2026Updated 2 months ago