Curated list of JAX Resources and Packages
☆55Oct 3, 2026Updated last month
Alternatives and similar repositories for awesome-jax
Users that are interested in awesome-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A project to illustrate the power of Jax for simulating astrochemical kinetics.☆11Oct 12, 2026Updated last month
- ☆16Dec 20, 2024Updated last year
- Deep neural network for solving continuous time economic models☆14Sep 17, 2026Updated last month
- Interpolation and function approximation with JAX☆283Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆541Updated 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.
- Optim4RL is a Jax framework of learning to optimize for reinforcement learning.☆28Nov 27, 2024Updated last year
- Chakra et_replay is a tool designed for replaying Chakra Execution Traces (ET) from machine learning models using PyTorch.☆14Oct 23, 2026Updated 2 weeks ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆586Updated this week
- Repository for the "Deep Learning in the Sequence Space" paper by Marlon Azinovic-Yang and Jan Zemlicka.☆19Apr 16, 2026Updated 6 months ago
- JAX translation of boltz☆32Aug 11, 2026Updated 3 months ago
- Issue tracker for https://alphaxiv.org☆25Oct 13, 2025Updated last year
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆48Sep 10, 2026Updated 2 months ago
- flexible VersionNumber parsing in Julia☆14May 16, 2023Updated 3 years ago
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Thermodynamic property models for use with Modia and other Julia packages☆10Aug 7, 2022Updated 4 years ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,999Oct 25, 2026Updated 2 weeks ago
- ☆10Sep 7, 2021Updated 5 years ago
- A an MLJ extension for accessing models and tools related to text analysis☆12Nov 6, 2025Updated last year
- Probabilistic solvers for differential equations in JAX.☆67Nov 5, 2026Updated last week
- calling GNU Octave functions from the Julia language☆11Jan 31, 2025Updated last year
- Regularized discriminant analysis in Julia.☆10Jan 10, 2021Updated 5 years ago
- Coordinates in JAX☆38Nov 3, 2026Updated last week
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆31Jul 18, 2023Updated 3 years ago
- 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.
- Reactive environments for self-learning agents☆25Nov 4, 2025Updated last year
- Traits for julia tables☆13Aug 31, 2026Updated 2 months ago
- Some standard pieces I find useful in Typst: fonts, subfigures, algorithms, labeled enums, etc.☆13Oct 16, 2026Updated 3 weeks ago
- Add a tqdm progress bar to your JAX scans and loops.☆131Oct 30, 2026Updated last week
- The CocoNuts project is an open-source humanoid robotics research platform. Its primary design philosophy is the minimization of leg iner…☆27Jun 30, 2026Updated 4 months ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆326Oct 14, 2026Updated 3 weeks ago
- Fast and reliable distributed systems in Python☆36Oct 25, 2026Updated 2 weeks ago
- OSCAR: configure and debug variational quantum algorithms efficiently☆10Sep 6, 2024Updated 2 years ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆234Nov 1, 2026Updated last week
- 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.
- An implementation of ESM2 in Equinox+JAX☆37Apr 20, 2026Updated 6 months ago
- Diffusion Probabilistic Model in Jax☆14Apr 20, 2024Updated 2 years ago
- Generates (almost) evenly distributed, equidistant points across a perfect sphere or the globe☆10Jun 30, 2018Updated 8 years ago
- GUI for interactive statistical analysis in Julia☆13Feb 8, 2020Updated 6 years ago
- Scientific Programming IRL: An intro to scientific programming concepts by scientists, for scientists☆12May 9, 2023Updated 3 years ago
- Providing probability distributions and non-negative measures over finite sets, whose elements are labelled.☆15Jul 1, 2026Updated 4 months ago
- ☆12Jul 27, 2026Updated 3 months ago
Are these results useful?