Curated list of JAX Resources and Packages
☆52Oct 3, 2026Updated last week
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:
- Optim4RL is a Jax framework of learning to optimize for reinforcement learning.☆28Nov 27, 2024Updated last year
- Orbax provides common checkpointing and persistence utilities for JAX users☆536Updated this week
- Interpolation and function approximation with JAX☆281Updated this week
- JAX translation of boltz☆32Aug 11, 2026Updated last month
- string diagrams for the working programmer☆16Jul 17, 2023Updated 3 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- A toolbox for simulating open quantum system dynamics.☆11Mar 21, 2022Updated 4 years ago
- Coordinates in JAX☆38Updated this week
- flexible VersionNumber parsing in Julia☆14May 16, 2023Updated 3 years ago
- Thermodynamic property models for use with Modia and other Julia packages☆10Aug 7, 2022Updated 4 years ago
- MLJ Interface for ScikitLearn.jl☆14May 22, 2024Updated 2 years ago
- ☆10Sep 7, 2021Updated 5 years ago
- Extension of dynamax repository to dynamical systems with continuous-time dynamics and measurements sampled at possibly irregular discret…☆24Oct 2, 2026Updated last week
- Sequences from Adaptyv Bio’s EGFR Protein Design Competition☆16Aug 28, 2025Updated last year
- calling GNU Octave functions from the Julia language☆11Jan 31, 2025Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Fast and reliable distributed systems in Python☆36Mar 25, 2026Updated 6 months ago
- Solving Tic-Tac-Toe with Neural Networks.☆21Jan 28, 2019Updated 7 years ago
- The CocoNuts project is an open-source humanoid robotics research platform. Its primary design philosophy is the minimization of leg iner…☆24Jun 30, 2026Updated 3 months ago
- Regularized discriminant analysis in Julia.☆10Jan 10, 2021Updated 5 years ago
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆31Jul 18, 2023Updated 3 years ago
- A package for wrapping iterative MLJ models in a control strategy☆12Sep 21, 2026Updated 2 weeks ago
- Traits for julia tables☆13Aug 31, 2026Updated last month
- An implementation of ESM2 in Equinox+JAX☆37Apr 20, 2026Updated 5 months ago
- A package to typeset ZX calculus directly in LaTeX.☆18Mar 15, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Providing probability distributions and non-negative measures over finite sets, whose elements are labelled.☆15Jul 1, 2026Updated 3 months ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆326Updated this week
- GUI for interactive statistical analysis in Julia☆13Feb 8, 2020Updated 6 years ago
- Julia implementation of the Varpro optimization algorithm☆15Apr 14, 2026Updated 5 months ago
- A JAX library for Sequential Convex Programming (SCP) Model Predictive Control (MPC) path planning☆36Jan 22, 2024Updated 2 years ago
- Streamlines bug reporting for julia☆20Feb 25, 2026Updated 7 months ago
- ☆36Nov 22, 2024Updated last year
- Python library for SBGN☆12Updated this week
- A complete example of batched refactorization in cuSOLVER.☆12Dec 22, 2021Updated 4 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.
- Schedule free optimiser implemented in JAX using Optimistix☆14May 29, 2024Updated 2 years ago
- Quantum computing tutorial held in Argonne National Laboratory in June of 2023☆13Jun 30, 2023Updated 3 years ago
- Python implementation of Parker wind models for exoplanet atmospheres☆16Sep 21, 2026Updated 2 weeks ago
- Code for the paper "Deep FTRL-ORW: An Efficient Deep Reinforcement Learning Algorithm for Solving Imperfect Information Extensive-Form Ga…☆11Dec 1, 2022Updated 3 years ago
- Drill hole utilities for the GeoStats.jl framework☆12Sep 15, 2026Updated 3 weeks ago
- ☆12Jun 14, 2022Updated 4 years ago
- ☆16Sep 4, 2025Updated last year