Modular, scalable library to train ML models
☆303Sep 24, 2026Updated this week
Alternatives and similar repositories for kauldron
Users that are interested in kauldron are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Library for reading and processing ML training data.☆782Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆533Updated this week
- ☆311Updated this week
- ☆389Updated this week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆561Sep 16, 2026Updated last week
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- a Jax quantization library☆136Updated this week
- Minimal yet performant LLM examples in pure JAX☆284Aug 23, 2026Updated last month
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆374Sep 15, 2026Updated last week
- A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax☆23Jun 8, 2025Updated last year
- A simple, performant, and scalable Jax LLM!☆2,428Updated this week
- CLU lets you write beautiful training loops in JAX.☆371Jul 7, 2026Updated 2 months ago
- Code for the "Binding via Reconstruction Clustering" paper☆21Jan 19, 2016Updated 10 years ago
- Minimal, lightweight JAX implementations of popular models.☆245May 29, 2026Updated 3 months ago
- PIX is an image processing library in JAX, for JAX.☆448Sep 9, 2026Updated 2 weeks ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Gemma open-weight LLM library, from Google DeepMind☆5,746Sep 16, 2026Updated last week
- A Lightweight LLM Post-Training Library☆2,460Updated this week
- JAX bindings for the flash-attention3 kernels☆23Sep 9, 2026Updated 2 weeks ago
- JAX-Toolbox☆433Updated this week
- Optax is a gradient processing and optimization library for JAX.☆2,342Updated this week
- ☆128Sep 5, 2026Updated 2 weeks ago
- Pytorch-like dataloaders for JAX.☆105Dec 16, 2025Updated 9 months ago
- Minimal but scalable implementation of large language models in JAX☆34Nov 28, 2025Updated 9 months ago
- Tokamax: A GPU and TPU kernel library.☆303Updated this week
- 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.
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated last year
- ☆143Updated this week
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,900Jun 22, 2025Updated last year
- A simple, performant and scalable JAX-based world modeling codebase.☆164Jan 15, 2026Updated 8 months ago
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆475Jun 18, 2026Updated 3 months ago
- ☆200Sep 16, 2026Updated last week
- ☆165Updated this week
- A simple library for scaling up JAX programs☆149Nov 4, 2025Updated 10 months ago
- Parallel hyperparameter tuning with JAX☆40Jul 18, 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.
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆460Jan 5, 2026Updated 8 months ago
- ☆362Sep 16, 2026Updated last week
- A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…☆155Oct 16, 2025Updated 11 months ago
- Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆16May 10, 2024Updated 2 years ago
- jax-triton contains integrations between JAX and OpenAI Triton☆470Updated this week
- Implementation of Forward Laplacian algorithm in JAX☆105Sep 15, 2026Updated last week
- FID computation in Jax/Flax.☆29Jul 17, 2024Updated 2 years ago