Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
☆46Jul 17, 2024Updated 2 years ago
Alternatives and similar repositories for out-of-the-box-fp8-training
Users that are interested in out-of-the-box-fp8-training 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 library for unit scaling in PyTorch☆137Jul 11, 2025Updated last year
- ☆16Apr 26, 2023Updated 3 years ago
- This is a simple torch implementation of the high performance Multi-Query Attention☆16Aug 23, 2023Updated 3 years ago
- ☆25Dec 11, 2024Updated last year
- Generic floating-point types in Python☆20Oct 18, 2026Updated last 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 repo based on XiLin Li's PSGD repo that extends some of the experiments.☆13Oct 7, 2024Updated 2 years ago
- Implementation of SelfExtend from the paper "LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning" from Pytorch and Zeta☆13Nov 11, 2024Updated last year
- Flexibly track outputs and grad-outputs of torch.nn.Module.☆13Oct 6, 2023Updated 3 years ago
- Conveniently export torch.compile compiled products into self-contained Python files☆35Jun 5, 2026Updated 4 months ago
- M4 experiment logbook☆59Aug 21, 2023Updated 3 years ago
- Implementation of the LDP module block in PyTorch and Zeta from the paper: "MobileVLM: A Fast, Strong and Open Vision Language Assistant …☆15Mar 11, 2024Updated 2 years ago
- Lowering PyTorch's Memory Consumption for Selective Differentiation☆12Aug 29, 2024Updated 2 years ago
- Code for DASH, a faster alternative to Distributed Shampoo via block stacking.☆13Feb 18, 2026Updated 8 months ago
- Microsoft Automatic Mixed Precision Library☆634Dec 1, 2025Updated 10 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- This document aims to agree on a broad, international strategy for the implementation of open scholarship that meets the needs of differe…☆11Dec 10, 2018Updated 7 years ago
- ☆13Jan 14, 2026Updated 9 months ago
- Codes and files for the paper Are Emergent Abilities in Large Language Models just In-Context Learning☆33Jan 9, 2025Updated last year
- Simple Implementation of a Transformer in the new framework MLX by Apple☆19Nov 18, 2024Updated last year
- ☆13Nov 14, 2024Updated last year
- ☆14Jun 24, 2024Updated 2 years ago
- ☆47Aug 15, 2019Updated 7 years ago
- My personal implementation of the model from "Qwen-VL: A Frontier Large Vision-Language Model with Versatile Abilities", they haven't rel…☆13Jan 29, 2024Updated 2 years ago
- Backup of the sources for my SJPO Teaching Notes☆11Apr 15, 2019Updated 7 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- AdamW optimizer for bfloat16 models in pytorch 🔥.☆41Jun 16, 2024Updated 2 years ago
- Import an ICS file into org to include events in your org agenda☆19Mar 30, 2026Updated 7 months ago
- ☆126May 28, 2024Updated 2 years ago
- Database for International Physics Olympiads☆13Aug 3, 2026Updated 2 months ago
- The project proposal template for OpenBioML community projects.☆18Feb 9, 2023Updated 3 years ago
- This repo is for reproducing our results in “Lipschitz Generative Adversarial Nets”.☆11Sep 26, 2020Updated 6 years ago
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 3 years ago
- ☆12Apr 26, 2024Updated 2 years ago
- Graph neural networks in JAX.☆67Jun 18, 2024Updated 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.
- Index of URLs to pdf files all over the internet and scripts☆25May 2, 2023Updated 3 years ago
- Code Roberta version of RetroMAE: Pre-Training Retrieval-oriented Language Models Via Masked Auto-Encoder☆10Mar 16, 2023Updated 3 years ago
- Implementation of Direct Preference Optimization☆17Jul 17, 2023Updated 3 years ago
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated 2 years ago
- An attempt at a SVD inpainting pipeline☆50Dec 24, 2023Updated 2 years ago
- Entailment self-training☆27May 30, 2023Updated 3 years ago
- Multi-Modal Multi-Embodied Hivemind-like Iteration of RTX-2☆15Jun 27, 2025Updated last year