topal-team / rockmateLinks
☆36Updated 8 months ago
Alternatives and similar repositories for rockmate
Users that are interested in rockmate are comparing it to the libraries listed below
Sorting:
- ☆10Updated 3 years ago
- Memory Optimizations for Deep Learning (ICML 2023)☆105Updated last year
- Fast Hadamard transform in CUDA, with a PyTorch interface☆224Updated last year
- ☆213Updated 2 years ago
- Distributed K-FAC preconditioner for PyTorch☆89Updated this week
- ☆159Updated last year
- ☆110Updated last year
- ☆29Updated 9 months ago
- Butterfly matrix multiplication in PyTorch☆174Updated last year
- This repository contains the experimental PyTorch native float8 training UX☆224Updated last year
- pytorch-profiler☆51Updated 2 years ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆271Updated last week
- Sparsity support for PyTorch☆36Updated 5 months ago
- Collection of kernels written in Triton language☆152Updated 4 months ago
- ☆51Updated last year
- ☆41Updated 4 years ago
- extensible collectives library in triton☆88Updated 5 months ago
- ☆42Updated 2 years ago
- A block oriented training approach for inference time optimization.☆34Updated last year
- ☆174Updated last year
- ☆32Updated last year
- Official PyTorch Implementation of HELP: Hardware-adaptive Efficient Latency Prediction for NAS via Meta-Learning (NeurIPS 2021 Spotlight…☆63Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆110Updated 3 months ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆133Updated 3 years ago
- ☆43Updated last year
- PyTorch extension for emulating FP8 data formats on standard FP32 Xeon/GPU hardware.☆110Updated 8 months ago
- A bunch of kernels that might make stuff slower 😉☆58Updated last week
- Complete GPU residency for ML.☆38Updated last month
- [ICML 2024 Oral] Any-Precision LLM: Low-Cost Deployment of Multiple, Different-Sized LLMs☆113Updated last month
- ☆328Updated this week