Slide decks, coding exercises, and quick references for learning the JAX AI Stack
☆264Jan 18, 2026Updated 3 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 10 months ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆15Updated this week
- ☆17Mar 4, 2025Updated last year
- Understand what physics/algorithms do transformers learn internally when trained on planetary motion☆41Feb 9, 2026Updated 2 months ago
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆19Jul 13, 2023Updated 2 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- The training codes of Jasper-Token-Compression-600M☆19Nov 19, 2025Updated 5 months ago
- The official code of "PixelWorld: Towards Perceiving Everything as Pixels" [TMLR25]☆16Sep 12, 2025Updated 7 months ago
- Package for Neural Posterior Estimation and Importance Sampling focused on Astronomical Applications☆40Aug 12, 2025Updated 8 months ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Mar 10, 2024Updated 2 years ago
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆40Apr 14, 2026Updated 3 weeks ago
- State-of-the-art paired encoder and decoder models (17M-1B params)☆69Aug 6, 2025Updated 8 months ago
- ☆10Sep 13, 2022Updated 3 years ago
- State of the art inference for your bayesian models.☆240Jan 14, 2026Updated 3 months ago
- Package of Pathways-on-Cloud utilities☆27Updated this week
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Source code for MA4270: Data Modelling and Computation on Transformers and Nadaraya-Watson Kernel Regression☆19May 29, 2024Updated last year
- Continuously tempered Hamiltonian Monte Carlo☆12Apr 12, 2017Updated 9 years ago
- Active Inference & Category Theory☆10Mar 11, 2024Updated 2 years ago
- Code to reproduce the material covered in Kùzu's YouTube tutorials☆20Oct 10, 2025Updated 6 months ago
- torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…☆218Apr 28, 2026Updated last week
- Probabilistic modeling of tabular data with normalizing flows.☆58Feb 11, 2026Updated 2 months ago
- ☆291Apr 27, 2026Updated last week
- The Scripps-KAUST Regional Integrated Prediction System (SKRIPS)☆22Feb 10, 2026Updated 2 months ago
- Kinematic and dynamic models of continuum and articulated soft robots.☆16Nov 22, 2025Updated 5 months 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.
- ☆55Apr 22, 2026Updated last week
- Material for the SC22 Deep Learning at Scale Tutorial☆41Jul 14, 2023Updated 2 years ago
- Winning solution for the TopCoW Challenge @MICCAI 2024☆18Nov 21, 2024Updated last year
- ☆47Feb 12, 2026Updated 2 months ago
- Fast, lightweight and parallelised simulation-based inference in JAX.☆23Oct 28, 2025Updated 6 months ago
- [NAACL 2024] A Framework aims to wisely initialize unseen subword embeddings in PLMs for efficient large-scale continued pretraining☆18Nov 26, 2023Updated 2 years ago
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆431Jan 5, 2026Updated 4 months ago
- [WACV'25] Official implementation of "PK-YOLO: Pretrained Knowledge Guided YOLO for Brain Tumor Detection in Multiplane MRI Slices".☆18Dec 19, 2025Updated 4 months ago
- ☆18Updated 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.
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆20Feb 8, 2024Updated 2 years ago
- Code for designing biased protein states☆79Apr 15, 2026Updated 2 weeks ago
- GUI for running simulations with universal MLIPs (MACE, CHGNet, SevenNet, Nequix, ORB, MatterSim, UPET, GRACE)☆31Updated this week
- автоответчик на отзывы на маркетплейсах☆17Feb 13, 2026Updated 2 months ago
- My CV☆40Updated this week
- Train text generation model with JavaScript.☆15Jul 14, 2024Updated last year
- A zero-config OpenAI client with support for 20+ providers, API key rotation, rate limits, optional LangChain integration and more.☆19Dec 11, 2025Updated 4 months ago