foundation-model-stack / fms-accelerationLinks
π Collection of libraries used with fms-hf-tuning to accelerate fine-tuning and training of large models.
β13Updated last week
Alternatives and similar repositories for fms-acceleration
Users that are interested in fms-acceleration are comparing it to the libraries listed below
Sorting:
- Experimental scripts for researching data adaptive learning rate scheduling.β22Updated 2 years ago
- Utilities for Training Very Large Modelsβ58Updated last year
- Source-to-Source Debuggable Derivatives in Pure Pythonβ15Updated last year
- Intel Gaudi's Megatron DeepSpeed Large Language Models for trainingβ13Updated 9 months ago
- Using FlexAttention to compute attention with different masking patternsβ46Updated last year
- β46Updated last year
- PyTorch centric eager mode debuggerβ48Updated 10 months ago
- Train, tune, and infer Bamba modelβ134Updated 4 months ago
- β21Updated 7 months ago
- β26Updated last year
- CUDA implementation of autoregressive linear attention, with all the latest research findingsβ44Updated 2 years ago
- Fork of Flame repo for training of some new stuff in developmentβ18Updated last week
- Code for the examples presented in the talk "Training a Llama in your backyard: fine-tuning very large models on consumer hardware" givenβ¦β14Updated 2 years ago
- DPO, but faster πβ45Updated 10 months ago
- [Oral; Neurips OPT2024 ] ΞΌLO: Compute-Efficient Meta-Generalization of Learned Optimizersβ13Updated 7 months ago
- β34Updated last year
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.β45Updated last year
- Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwiseβ39Updated last year
- Here we will test various linear attention designs.β61Updated last year
- A collection of reproducible inference engine benchmarksβ34Updated 5 months ago
- JAX Scalify: end-to-end scaled arithmeticsβ16Updated 11 months ago
- β15Updated last year
- β22Updated 10 months ago
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inferenceβ68Updated 10 months ago
- Official code for the paper "Attention as a Hypernetwork"β43Updated last year
- IBM development fork of https://github.com/huggingface/text-generation-inferenceβ61Updated last month
- TorchFix - a linter for PyTorch-using code with autofix supportβ149Updated last month
- Example ML projects that use the Determined library.β32Updated last year
- Automatically take good care of your preemptible TPUsβ37Updated 2 years ago
- ML/DL Math and Method notesβ64Updated last year