☆151Feb 12, 2026Updated 2 weeks ago
Alternatives and similar repositories for saxml
Users that are interested in saxml are comparing it to the libraries listed below
Sorting:
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆549Updated this week
- ☆192Feb 16, 2026Updated last week
- ☆11Mar 1, 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…☆413Jan 5, 2026Updated last month
- ☆16Feb 18, 2026Updated last week
- xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat…☆169Feb 20, 2026Updated last week
- ☆18Jul 9, 2024Updated last year
- ☆13Feb 12, 2026Updated 2 weeks ago
- ☆348Feb 19, 2026Updated last week
- Orbax provides common checkpointing and persistence utilities for JAX users☆482Updated this week
- ☆135Dec 22, 2025Updated 2 months ago
- jax-triton contains integrations between JAX and OpenAI Triton☆439Updated this week
- ☆12Mar 10, 2023Updated 2 years ago
- Library for reading and processing ML training data.☆685Updated this week
- ☆367Apr 12, 2024Updated last year
- A simple, performant and scalable Jax LLM!☆2,148Updated this week
- ☆14Jul 15, 2024Updated last year
- ☆16Oct 20, 2025Updated 4 months ago
- Task-based datasets, preprocessing, and evaluation for sequence models.☆594Feb 3, 2026Updated 3 weeks ago
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆79Dec 18, 2025Updated 2 months ago
- ☆20Jul 5, 2024Updated last year
- ☆61Updated this week
- ☆314Updated this week
- Minimal yet performant LLM examples in pure JAX☆240Jan 14, 2026Updated last month
- ☆17Mar 17, 2023Updated 2 years ago
- Collection of eclectic utils for python.☆252Feb 20, 2026Updated last week
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆106Jun 28, 2025Updated 8 months ago
- ☆18Feb 18, 2026Updated last week
- ☆376Feb 14, 2026Updated 2 weeks ago
- JAX implementation of the Llama 2 model☆216Feb 2, 2024Updated 2 years ago
- GCE Rescue is a command-line tool to boot Google Cloud Platform VMs in Rescue Mode.☆14Dec 22, 2023Updated 2 years ago
- ☆566Jul 11, 2024Updated last year
- ☆12Feb 19, 2026Updated last week
- [WACV 2026] PyTorch code for 4D-Animal.☆27Nov 18, 2025Updated 3 months ago
- ☆13Dec 10, 2025Updated 2 months ago
- ☆13Jun 18, 2024Updated last year
- ☆15May 11, 2025Updated 9 months ago
- ☆18Jun 18, 2025Updated 8 months ago
- 🚀 Collection of libraries used with fms-hf-tuning to accelerate fine-tuning and training of large models.☆13Jan 30, 2026Updated last month