Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
☆19Jul 24, 2025Updated last year
Alternatives and similar repositories for llm-jax
Users that are interested in llm-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:
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- Automatically take good care of your preemptible TPUs☆37May 15, 2023Updated 3 years ago
- ☆19Sep 2, 2026Updated last month
- a Jax/Flax inference code of StarCoder☆12Jun 12, 2023Updated 3 years ago
- ☆32Nov 4, 2024Updated last year
- 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.
- ☆22Dec 15, 2023Updated 2 years ago
- ☆23Jan 5, 2025Updated last year
- An AI character interaction system with emotional modeling and advanced memory management☆17Aug 6, 2026Updated 2 months ago
- ☆22Nov 18, 2024Updated last year
- implementation of https://arxiv.org/pdf/2312.09299☆21Jul 3, 2024Updated 2 years ago
- ☆23Dec 16, 2024Updated last year
- (EasyDel Former) is a utility library designed to simplify and enhance the development in JAX☆33Updated this week
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆207May 30, 2026Updated 4 months ago
- Efficient optimizers☆340Sep 9, 2026Updated last month
- 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.
- Machine Learning eXperiment Utilities☆48Jul 29, 2025Updated last year
- ☆19Dec 4, 2025Updated 10 months ago
- ☆13Apr 25, 2024Updated 2 years ago
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated last year
- CIFAR10 ResNets implemented in JAX+Flax☆12Apr 6, 2022Updated 4 years ago
- Tool for generating pictures using mathematical formulas.☆44Nov 8, 2021Updated 4 years ago
- A fully trainable state space model (SSM)☆16Mar 18, 2025Updated last year
- Bullseye Polytope Clean-Label Poisoning Attack☆18Nov 5, 2020Updated 5 years ago
- Train to 94% on CIFAR-10 in 4.4 seconds on a single A100☆12Dec 30, 2023Updated 2 years ago
- AI Agents on 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 simple library for scaling up JAX programs☆149Nov 4, 2025Updated 11 months ago
- ☆12May 30, 2025Updated last year
- LLM training in simple, raw C/CUDA☆15Dec 5, 2024Updated last year
- ☆18Aug 24, 2024Updated 2 years ago
- Evaluating language models on word puzzle games☆10Oct 25, 2024Updated last year
- A PyTorch implementation of the Exclusive Cross Entropy Loss.☆20Aug 12, 2022Updated 4 years ago
- Neural network density models for speech separation.☆20Nov 26, 2020Updated 5 years ago
- Orthogonal Matching Pursuit, parallelized on both CPU and GPU. 100x+ Speedup☆18Oct 4, 2026Updated last week
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆32Jul 28, 2023Updated 3 years ago
- 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.
- ☆17Jun 22, 2025Updated last year
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆375Sep 26, 2026Updated 3 weeks ago
- VeLO optimizer in PyTorch☆20Feb 6, 2023Updated 3 years ago
- Lightweight package that tracks and summarizes code changes using LLMs (Large Language Models)☆32Feb 27, 2025Updated last year
- Deploy KoGPT with Triton Inference Server☆14Nov 18, 2022Updated 3 years ago
- Using multiple LLMs for ensemble Forecasting☆16Jan 17, 2024Updated 2 years ago
- YOLOv10: Real-Time End-to-End Object Detection☆12May 24, 2024Updated 2 years ago