Slide decks, coding exercises, and quick references for learning the JAX AI Stack
☆277May 22, 2026Updated 2 months ago
Alternatives and similar repositories for Learning-JAX
Users that are interested in Learning-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 collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax☆23Jun 8, 2025Updated last year
- Minimal, lightweight JAX implementations of popular models.☆240May 29, 2026Updated 2 months ago
- Interactive exploration of equivariant neural networks on homogeneous spaces, with a focus on the sphere S² as SO(3)/SO(2). From Lecture …☆16May 6, 2025Updated last year
- ☆23May 26, 2026Updated 2 months ago
- Package for data-driven and phenomenological gravitational waveform models☆12Updated this week
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- GCM Physics written in JAX☆104Updated this week
- A python library for self-supervised learning☆12Nov 17, 2025Updated 8 months ago
- ☆121Updated this week
- ☆309Updated this week
- Library for reading and processing ML training data.☆762Updated this week
- [ICCV'25] ViLU: Learning Vision-Language Uncertainties for Failure Prediction☆16Jul 16, 2025Updated last year
- ☆379Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆527Updated this week
- ☆33Dec 18, 2023Updated 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.
- State of the art inference for your bayesian models.☆248Jan 14, 2026Updated 6 months ago
- JAX-Toolbox☆425Updated this week
- Minimal copilot for your terminal shell, supports 8+ languages, OpenAI + Bring your own model☆16Jun 21, 2026Updated last month
- Minimal yet performant LLM examples in pure JAX☆271Jul 4, 2026Updated last month
- ☆16Jul 6, 2026Updated last month
- Sequential Monte Carlo with NUTS for Gravitational Wave Inference☆21Jun 18, 2026Updated last month
- JAX - A curated list of resources https://github.com/google/jax☆2,144Jan 20, 2026Updated 6 months ago
- Gravitational-wave astronomy in JAX☆15Jul 24, 2026Updated 2 weeks ago
- Recipes for reproducing training and serving benchmarks for large machine learning models using GPUs on Google Cloud.☆140Jul 28, 2026Updated last week
- 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.
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆38Apr 21, 2025Updated last year
- ☆30Oct 24, 2025Updated 9 months ago
- FLS point cloud registration (ISRR 2022)☆19Jun 12, 2024Updated 2 years ago
- ☆37Jul 27, 2025Updated last year
- JAX-accelerated nuclear equation of state inference and TOV solvers☆16Jul 24, 2026Updated 2 weeks ago
- Example of GAMES (Gravitational-wave Amortized Metric Enhanced Sampler)☆12Oct 8, 2024Updated last year
- ☆591Jul 11, 2024Updated 2 years ago
- Astronomy Research + JAX Meeting 2024☆13Oct 23, 2024Updated last year
- Lorenz 1996 two time-scale model for learning machine learning☆62May 25, 2026Updated 2 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Conditional Flow Matching using JAX☆13Jun 22, 2023Updated 3 years ago
- FlexiTokens☆23Dec 27, 2025Updated 7 months ago
- torchprime is a reference model implementation for PyTorch on TPU.☆49Mar 3, 2026Updated 5 months ago
- Clean version of research-projects, just with ILE☆18Updated this week
- ☆110Jun 2, 2025Updated last year
- A Multilingual Keyboard Layout-Based Typo Generator☆17Nov 23, 2025Updated 8 months ago
- A collection of robotics problems, for learning and for fun☆16Jul 19, 2026Updated 3 weeks ago