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:
- ☆16Apr 26, 2023Updated 3 years ago
- A library for unit scaling in PyTorch☆136Jul 11, 2025Updated last year
- This is a simple torch implementation of the high performance Multi-Query Attention☆16Aug 23, 2023Updated 2 years ago
- A repo based on XiLin Li's PSGD repo that extends some of the experiments.☆14Oct 7, 2024Updated last year
- M4 experiment logbook☆59Aug 21, 2023Updated 2 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.
- Flexibly track outputs and grad-outputs of torch.nn.Module.☆13Oct 6, 2023Updated 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.☆35Feb 3, 2026Updated 6 months ago
- AdamW optimizer for bfloat16 models in pytorch 🔥.☆41Jun 16, 2024Updated 2 years ago
- Simple Implementation of a Transformer in the new framework MLX by Apple☆19Nov 18, 2024Updated last year
- Implicit generative models and related stuff based on the MMD, in PyTorch☆16Sep 24, 2020Updated 5 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
- ☆47Aug 15, 2019Updated 6 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.
- Code associated to papers on superposition (in ML interpretability)☆40Sep 13, 2022Updated 3 years ago
- Microsoft Automatic Mixed Precision Library☆635Dec 1, 2025Updated 8 months ago
- 📀 Encoding YUV into H.246 & transmit it to rtmp server.☆10Dec 22, 2023Updated 2 years ago
- Graph neural networks in JAX.☆67Jun 18, 2024Updated 2 years ago
- Variational Autoencoder with non-euclidean (hyperbolic) latent space☆14Nov 25, 2022Updated 3 years ago
- Tabular data imputation and generation, with flexible modeling of quantitative features via hierarchical binning (TMLR, 2025)☆16Mar 10, 2025Updated last year
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- Fast finetuning using a booster model that puts the initial state to a local minimum☆113Aug 29, 2023Updated 2 years ago
- The project proposal template for OpenBioML community projects.☆18Feb 9, 2023Updated 3 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.
- ☆14Mar 8, 2025Updated last year
- Jax Decompiler☆16Apr 22, 2025Updated last year
- Entailment self-training☆27May 30, 2023Updated 3 years ago
- Multi-Modal Multi-Embodied Hivemind-like Iteration of RTX-2☆15Jun 27, 2025Updated last year
- Public repository for managing Grid Platform documentation synced with gitbook on docs.grid.ai☆20Mar 27, 2026Updated 4 months ago
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- ☆16Dec 30, 2024Updated last year
- 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
- Lossless compression using Probabilistic Circuits☆16Mar 10, 2022Updated 4 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.
- Synthetic Alphabet Dataset☆19Mar 27, 2025Updated last year
- Reference implementation for Reward-Augmented Decoding: Efficient Controlled Text Generation With a Unidirectional Reward Model☆45Oct 1, 2025Updated 10 months ago
- [IJCAI'23] Semantic-aware Generation of Multi-view Portrait Drawings (SAGE)☆10Feb 25, 2024Updated 2 years ago
- Tools for generating cubed sphere grids and solving partial differential equations on the sphere.☆15Jan 20, 2026Updated 6 months ago
- Fast singularity detection with kernel☆39Jan 4, 2024Updated 2 years ago
- ☆14Apr 7, 2022Updated 4 years ago
- JAX for Graphcore IPU (experimental)☆21Mar 12, 2024Updated 2 years ago