saifhaq / alma
☆19Updated last month
Alternatives and similar repositories for alma
Users that are interested in alma are comparing it to the libraries listed below
Sorting:
- Accelerated First Order Parallel Associative Scan☆182Updated 8 months ago
- Cataloging released Triton kernels.☆221Updated 4 months ago
- A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS☆173Updated last week
- Implementation of a Transformer, but completely in Triton☆265Updated 3 years ago
- ☆155Updated last year
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 9 months ago
- ☆204Updated 3 weeks ago
- Collection of kernels written in Triton language☆122Updated last month
- ☆163Updated 4 months ago
- Implementation of Flash Attention in Jax☆209Updated last year
- ☆88Updated last year
- Experiment of using Tangent to autodiff triton☆78Updated last year
- A repository for log-time feedforward networks☆222Updated last year
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆45Updated 10 months ago
- A bunch of kernels that might make stuff slower 😉☆40Updated this week
- seqax = sequence modeling + JAX☆155Updated last month
- Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*☆82Updated last year
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers☆92Updated 10 months ago
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆132Updated last year
- ☆32Updated 4 months ago
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆180Updated this week
- ring-attention experiments☆142Updated 7 months ago
- supporting pytorch FSDP for optimizers☆80Updated 5 months ago
- Applied AI experiments and examples for PyTorch☆267Updated this week
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆121Updated 9 months ago
- ☆79Updated 10 months ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆195Updated this week
- The simplest but fast implementation of matrix multiplication in CUDA.☆35Updated 9 months ago
- Fast low-bit matmul kernels in Triton☆299Updated this week
- making the official triton tutorials actually comprehensible☆30Updated last month