Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
☆46Jul 17, 2024Updated last year
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:
- ☆16Apr 26, 2023Updated 3 years ago
- ☆24Dec 11, 2024Updated last year
- A library for unit scaling in PyTorch☆133Jul 11, 2025Updated 10 months ago
- A repo based on XiLin Li's PSGD repo that extends some of the experiments.☆14Oct 7, 2024Updated last year
- 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
- 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.
- 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
- Implementation of the LOSSGRAD optimization algorithm☆15Mar 21, 2019Updated 7 years ago
- Lowering PyTorch's Memory Consumption for Selective Differentiation☆12Aug 29, 2024Updated last year
- We study toy models of skill learning.☆33Feb 3, 2026Updated 4 months ago
- AdamW optimizer for bfloat16 models in pytorch 🔥.☆40Jun 16, 2024Updated last year
- BFloat16 Fused Adam Operator for PyTorch☆19Nov 16, 2024Updated last year
- Microsoft Automatic Mixed Precision Library☆637Dec 1, 2025Updated 6 months ago
- ☆125May 28, 2024Updated 2 years ago
- Utilities for working with formulas, expressions, calls and other R objects☆17Nov 18, 2018Updated 7 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Graph neural networks in JAX.☆67Jun 18, 2024Updated last year
- This repo is for reproducing our results in “Lipschitz Generative Adversarial Nets”.☆11Sep 26, 2020Updated 5 years ago
- ☆12Apr 26, 2024Updated 2 years ago
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- Tabular data imputation and generation, with flexible modeling of quantitative features via hierarchical binning (TMLR, 2025)☆16Mar 10, 2025Updated last year
- The project proposal template for OpenBioML community projects.☆18Feb 9, 2023Updated 3 years ago
- Self-Attention Generative Adversarial Networks with Relativistic loss☆13Feb 14, 2019Updated 7 years ago
- ☆14Apr 7, 2022Updated 4 years ago
- Multi-Modal Multi-Embodied Hivemind-like Iteration of RTX-2☆15Jun 27, 2025Updated 11 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Classifier for CIFAR-10. Grayscaling, HOG, PCA, and RBF SVM. 62% test accuracy. Walkthrough on YouTube: https://youtu.be/gmTweV0eHhk☆14Nov 24, 2024Updated last year
- Automatic High-Order Optimization for Tensors☆22Apr 14, 2023Updated 3 years ago
- Reference implementation for Reward-Augmented Decoding: Efficient Controlled Text Generation With a Unidirectional Reward Model☆45Oct 1, 2025Updated 8 months ago
- Synthetic Alphabet Dataset☆19Mar 27, 2025Updated last year
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆22Mar 23, 2026Updated 2 months ago
- ICML2019 Accepted Paper. Overcoming Multi-Model Forgetting☆14Jun 5, 2019Updated 7 years ago
- Minimalistic, hackable PyTorch implementation of SimSiam in ~400 lines. Achieves good performance on ImageNet with ResNet50. Features dis…☆22Nov 25, 2024Updated last year
- Finetune the 1.4B latent diffusion text2img-large checkpoint from CompVis using deepspeed. (work-in-progress)☆36Apr 17, 2022Updated 4 years ago
- [IJCAI'23] Semantic-aware Generation of Multi-view Portrait Drawings (SAGE)☆10Feb 25, 2024Updated 2 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- ☆11Apr 23, 2023Updated 3 years ago
- Generative cellular automaton-like learning environments for RL.☆20Jan 30, 2025Updated last year
- ☆68Dec 8, 2023Updated 2 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆30Sep 25, 2021Updated 4 years ago
- Continual Memorization of Factoids in Large Language Models☆12Nov 20, 2024Updated last year
- JAX interpreter for Vulkan☆17Jun 1, 2021Updated 5 years ago
- A swarm of LLM agents that will help you test, document, and productionize your code!☆19May 25, 2026Updated 2 weeks ago