Curated list of JAX Resources and Packages
☆51Aug 30, 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☆533Updated this week
- Multiple GEMM operators are constructed with cutlass to support LLM inference.☆20Aug 3, 2025Updated last year
- string diagrams for the working programmer☆15Jul 17, 2023Updated 3 years ago
- A toolbox for simulating open quantum system dynamics.☆11Mar 21, 2022Updated 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.
- 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
- Brax Viewer is a real-time, interactive web viewer for monitoring reinforcement learning (RL) policies☆17Nov 28, 2025Updated 9 months ago
- ☆10Sep 7, 2021Updated 5 years ago
- 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
- Fast and reliable distributed systems in Python☆35Mar 25, 2026Updated 5 months ago
- The CocoNuts project is an open-source humanoid robotics research platform. Its primary design philosophy is the minimization of leg iner…☆23Jun 30, 2026Updated 2 months ago
- Regularized discriminant analysis in Julia.☆10Jan 10, 2021Updated 5 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆31Jul 18, 2023Updated 3 years ago
- An implementation of ESM2 in Equinox+JAX☆36Apr 20, 2026Updated 4 months ago
- A package for wrapping iterative MLJ models in a control strategy☆12Aug 17, 2026Updated 3 weeks ago
- Traits for julia tables☆13Aug 31, 2026Updated last week
- Providing probability distributions and non-negative measures over finite sets, whose elements are labelled.☆15Jul 1, 2026Updated 2 months ago
- ☆18Jan 5, 2023Updated 3 years ago
- GUI for interactive statistical analysis in Julia☆13Feb 8, 2020Updated 6 years ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆328Aug 6, 2026Updated last month
- Project page for the NeurIPS 2024 paper, Language Grounded Multi-agent Reinforcement Learning with Human-interpretable Communication.☆18Dec 6, 2024Updated last year
- 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.
- Streamlines bug reporting for julia☆20Feb 25, 2026Updated 6 months ago
- ☆36Nov 22, 2024Updated last year
- A complete example of batched refactorization in cuSOLVER.☆12Dec 22, 2021Updated 4 years ago
- 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
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated 2 years ago
- A bare-metal Python library for building and manipulating the molecular structures of proteins, nucleic acids, and small organic molecule…☆20Updated this week
- 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
- EleutherAI ML Performance reading group repository (slides, meeting recordings, annotated papers)☆37Mar 20, 2026Updated 5 months ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Documentation pages for IQM's client-side libraries.☆17Updated this week
- Deep memory and sequence models in JAX☆32Jun 8, 2026Updated 2 months ago
- Discrete, Continuous, and Manifold Flow Matching with Splits and Deletions☆17Jun 22, 2026Updated 2 months ago
- ☆16Sep 4, 2025Updated last year
- Automatic compilation for Julia packages☆17Apr 10, 2023Updated 3 years ago
- A solver for linear complementarity problems☆12Dec 16, 2021Updated 4 years ago
- Score-based Diffusion models in JAX.☆19Dec 29, 2025Updated 8 months ago