topal-team / rockmateLinks
☆36Updated last year
Alternatives and similar repositories for rockmate
Users that are interested in rockmate are comparing it to the libraries listed below
Sorting:
- ☆160Updated 2 years ago
- ☆10Updated 3 years ago
- ☆222Updated 2 years ago
- Memory Optimizations for Deep Learning (ICML 2023)☆114Updated last year
- ☆115Updated last year
- A bunch of kernels that might make stuff slower 😉☆73Updated this week
- Collection of kernels written in Triton language☆174Updated 9 months ago
- A library for unit scaling in PyTorch☆133Updated 5 months ago
- Fast Hadamard transform in CUDA, with a PyTorch interface☆267Updated 2 months ago
- ☆39Updated 3 weeks ago
- This repository contains the experimental PyTorch native float8 training UX☆227Updated last year
- pytorch-profiler☆50Updated 2 years ago
- Distributed K-FAC preconditioner for PyTorch☆93Updated 2 weeks ago
- ☆31Updated last year
- Experiment of using Tangent to autodiff triton☆81Updated last year
- A block oriented training approach for inference time optimization.☆34Updated last year
- Butterfly matrix multiplication in PyTorch☆178Updated 2 years ago
- extensible collectives library in triton☆91Updated 9 months ago
- ☆114Updated 2 weeks ago
- ☆60Updated last year
- Training neural networks in TensorFlow 2.0 with 5x less memory☆137Updated 3 years ago
- Framework to reduce autotune overhead to zero for well known deployments.☆92Updated 3 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆139Updated 7 months ago
- The simplest but fast implementation of matrix multiplication in CUDA.☆39Updated last year
- Triton-based implementation of Sparse Mixture of Experts.☆259Updated 3 months ago
- ☆168Updated 2 years ago
- ☆43Updated last year
- ☆30Updated last year
- ☆84Updated 11 months ago
- ☆38Updated last year