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:
- ☆10Updated 3 years ago
- ☆160Updated 2 years ago
- Butterfly matrix multiplication in PyTorch☆178Updated 2 years ago
- ☆222Updated 2 years ago
- Collection of kernels written in Triton language☆175Updated 9 months ago
- ☆115Updated last year
- pytorch-profiler☆50Updated 2 years ago
- ☆31Updated last year
- Fast Hadamard transform in CUDA, with a PyTorch interface☆275Updated 3 months ago
- This repository contains the experimental PyTorch native float8 training UX☆227Updated last year
- Official PyTorch Implementation of HELP: Hardware-adaptive Efficient Latency Prediction for NAS via Meta-Learning (NeurIPS 2021 Spotlight…☆64Updated last year
- ☆40Updated last year
- Training neural networks in TensorFlow 2.0 with 5x less memory☆137Updated 3 years ago
- extensible collectives library in triton☆93Updated 10 months ago
- ☆43Updated 2 years ago
- A bunch of kernels that might make stuff slower 😉☆75Updated this week
- ☆27Updated 2 years ago
- Distributed K-FAC preconditioner for PyTorch☆94Updated this week
- Memory Optimizations for Deep Learning (ICML 2023)☆114Updated last year
- ☆39Updated last month
- Sparsity support for PyTorch☆38Updated 10 months ago
- Framework to reduce autotune overhead to zero for well known deployments.☆94Updated 4 months ago
- Research and development for optimizing transformers☆131Updated 4 years ago
- ☆45Updated 2 years ago
- PyTorch bindings for CUTLASS grouped GEMM.☆141Updated 8 months ago
- A library for unit scaling in PyTorch☆133Updated 6 months ago
- ☆186Updated last year
- ☆60Updated last year
- A block oriented training approach for inference time optimization.☆34Updated last year
- Flexible simulator for mixed precision and format simulation of LLMs and vision transformers.☆51Updated 2 years ago