topal-team / rockmateLinks
☆36Updated 5 months ago
Alternatives and similar repositories for rockmate
Users that are interested in rockmate are comparing it to the libraries listed below
Sorting:
- Memory Optimizations for Deep Learning (ICML 2023)☆64Updated last year
- ☆10Updated 3 years ago
- ☆43Updated last year
- ☆49Updated 10 months ago
- Sparsity support for PyTorch☆35Updated 2 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆127Updated this week
- A block oriented training approach for inference time optimization.☆33Updated 9 months ago
- pytorch-profiler☆51Updated 2 years ago
- A library for unit scaling in PyTorch☆125Updated 6 months ago
- Fast Hadamard transform in CUDA, with a PyTorch interface☆195Updated last year
- Official PyTorch Implementation of HELP: Hardware-adaptive Efficient Latency Prediction for NAS via Meta-Learning (NeurIPS 2021 Spotlight…☆63Updated 10 months ago
- PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021☆56Updated 3 years ago
- Experiment of using Tangent to autodiff triton☆79Updated last year
- ☆105Updated 9 months ago
- extensible collectives library in triton☆87Updated 2 months ago
- ☆25Updated 6 months ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆44Updated 10 months ago
- PyTorch extension for emulating FP8 data formats on standard FP32 Xeon/GPU hardware.☆110Updated 6 months ago
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆38Updated 2 years ago
- This repository contains the experimental PyTorch native float8 training UX☆223Updated 10 months ago
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆52Updated 4 years ago
- ☆10Updated 6 months ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆153Updated this week
- ☆206Updated 2 years ago
- Collection of kernels written in Triton language☆127Updated 2 months ago
- ☆59Updated last year
- Distributed K-FAC preconditioner for PyTorch☆87Updated last week
- PyTorch bindings for CUTLASS grouped GEMM.☆93Updated last week
- ☆30Updated last year
- A bunch of kernels that might make stuff slower 😉☆48Updated this week