A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.
☆129Dec 29, 2025Updated 10 months ago
Alternatives and similar repositories for JAXformer
Users that are interested in JAXformer are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- realistic obama☆12Sep 28, 2025Updated last year
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated last year
- A simple, performant and scalable JAX-based world modeling codebase.☆165Jan 15, 2026Updated 9 months ago
- Powerful Asynchronous Multi agent framework built from scratch in python supporting RAG and MCP compatibilities.☆12Aug 26, 2025Updated last year
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Minimal yet performant LLM examples in pure JAX☆286Aug 23, 2026Updated 2 months ago
- A collection of my from-scratch implementation of various models deployed on HF Spaces☆14Oct 14, 2025Updated last year
- ☆594Jul 11, 2024Updated 2 years ago
- A JAX implementation of stochastic addition.☆13Aug 15, 2022Updated 4 years ago
- MoE training for Me and You and maybe other people☆400Mar 15, 2026Updated 7 months ago
- Parallel hyperparameter tuning with JAX☆40Sep 30, 2026Updated last month
- ☆95Jul 5, 2024Updated 2 years ago
- Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)☆24Apr 17, 2024Updated 2 years ago
- Minimal, lightweight JAX implementations of popular models.☆246May 29, 2026Updated 5 months ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- ☆307Jul 15, 2024Updated 2 years ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Jul 24, 2025Updated last year
- ☆15Mar 20, 2025Updated last year
- Stitch your rollouts together across clouds and regions☆36Updated this week
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆1,513Oct 19, 2026Updated last week
- Minimal but scalable implementation of large language models in JAX☆34Nov 28, 2025Updated 11 months ago
- SpecTrax is a JAX-native library for neural networks and graph learning, built for performance, composability and modularity.☆44Updated this week
- Scripts and config files for SLURM☆13Sep 12, 2025Updated last year
- Graph neural networks in JAX.☆67Jun 18, 2024Updated 2 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.
- A simple, performant, and scalable Jax LLM!☆2,455Updated this week
- Einsum-like high-level array sharding API for JAX☆35Jul 16, 2024Updated 2 years ago
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated last year
- Jax/Flax rewrite of Karpathy's nanoGPT☆66Feb 15, 2023Updated 3 years ago
- Differentiable numerical solvers and scientific machine learning on one JAX substrate; you write the math, it stays differentiable end to…☆33Updated this week
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆718Jan 26, 2026Updated 9 months ago
- Custom triton kernels for training Karpathy's nanoGPT.☆19Oct 21, 2024Updated 2 years ago
- AI-native open-source finite-element platform connecting engineering, computation, data, and AI.☆33Updated this week
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆218Jun 1, 2025Updated last year
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Classes/structs, pattern matching, safe counters... and more! Your one-stop library for programming tools not already in core Typst.☆30Oct 16, 2026Updated 2 weeks ago
- https://github.com/gpu-mode/reference-kernels☆27Jul 4, 2026Updated 3 months ago
- A small, readable PyTorch inference engine for Qwen3-4B with continuous batching, native paged attention, prefix caching, and an OpenAI-c…☆18Updated this week
- Schedule free optimiser implemented in JAX using Optimistix☆14May 29, 2024Updated 2 years ago
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆83Oct 16, 2026Updated 2 weeks ago
- Programmable chat templates for LLM training and inference.☆167Updated this week
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated last year