Curated list of JAX Resources and Packages
☆35Mar 2, 2026Updated 3 weeks ago
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:
- Extension of dynamax repository to dynamical systems with continuous-time dynamics and measurements sampled at possibly irregular discret…☆20Updated this week
- An implementation of ESM2 in Equinox+JAX☆36Jun 5, 2025Updated 9 months ago
- Multiple GEMM operators are constructed with cutlass to support LLM inference.☆20Aug 3, 2025Updated 7 months ago
- Diffusion Probabilistic Model in Jax☆13Apr 20, 2024Updated last year
- Optim4RL is a Jax framework of learning to optimize for reinforcement learning.☆28Nov 27, 2024Updated last year
- 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.
- Implementation of the HTCPCP Protocol☆14Feb 3, 2013Updated 13 years ago
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Jan 7, 2025Updated last year
- Prototype for sparse automatic differentiation in JAX + Julia☆21Jun 17, 2025Updated 9 months ago
- Python library for SBGN☆11Mar 12, 2026Updated 2 weeks ago
- ☆35Nov 22, 2024Updated last year
- ☆13Dec 20, 2024Updated last year
- 🚀 Official code for “XStreamVGGT: Extremely Memory-Efficient Streaming Vision Geometry Grounded Transformer with KV Cache Compression”, …☆41Jan 27, 2026Updated 2 months ago
- Solving Tic-Tac-Toe with Neural Networks.☆19Jan 28, 2019Updated 7 years ago
- ECE250 Testcases and Autograder☆25Mar 28, 2023Updated 2 years ago
- 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.
- A complete example of batched refactorization in cuSOLVER.☆13Dec 22, 2021Updated 4 years ago
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated last year
- Polish datsets for grammatical error correction☆12Oct 13, 2023Updated 2 years ago
- ☆15Sep 4, 2025Updated 6 months ago
- Average-Reward Reinforcement Learning with Trust Region Methods☆11Oct 17, 2022Updated 3 years ago
- A curated list of Soft Robotics resources, projects, courses, books, video lectures, papers, journals and articles.☆16Mar 5, 2024Updated 2 years ago
- We introduce algebraic machine reasoning to solve Raven's Progressive Matrices (RPMs).☆15Jul 2, 2025Updated 8 months ago
- ☆12Jun 14, 2022Updated 3 years ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆53Jan 23, 2026Updated 2 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- [NeurIPS 2025] LabUtopia: High-Fidelity Simulation and Hierarchical Benchmark for Scientific Embodied Agents☆31Updated this week
- PolytopeWalk: fast sparse and dense random walks on polytope, with C++ backend and Python interface☆16Jul 25, 2025Updated 8 months ago
- Find, list, filter, and edit or otherwise act on files from the terminal using fuzzy-matching or regular expressions.☆26May 27, 2018Updated 7 years ago
- FORward MOdeling tool for Spectral Analysis☆11Updated this week
- 🏍️ - JAX-based framework to model biological systems☆17Jan 15, 2026Updated 2 months ago
- JAX code for modeling nearly-Keplerian orbits☆18Updated this week
- Python implementation of Parker wind models for exoplanet atmospheres☆13May 12, 2025Updated 10 months ago
- Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)☆16Oct 11, 2021Updated 4 years ago
- USD models for NVIDIA Isaac Sim☆16Aug 8, 2025Updated 7 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- A Python package for working with pipeline-produced spectra from IGRINS, HPF, and Keck NIRSPEC☆19Jan 12, 2026Updated 2 months ago
- Stochastic Gradient MCMC for Jax☆18May 19, 2025Updated 10 months ago
- An example of continuous integration and deployment of a sphinx project☆13Aug 25, 2023Updated 2 years ago
- JAX translation of boltz☆25Aug 4, 2025Updated 7 months ago
- SKFAC Preconditioner for MindSpore☆12Jul 2, 2021Updated 4 years ago
- Modern Python & C++ API for the radiative transfer solver DISORT. Parallelized with PyTorch tensors; compile-free with: pip install pydis…☆15Mar 1, 2026Updated 3 weeks ago
- Unofficial implementation of the Ask-LLM paper 'How to Train Data-Efficient LLMs', arXiv:2402.09668.☆12Jun 19, 2024Updated last year