topal-team / rockmateLinks
☆36Updated 11 months ago
Alternatives and similar repositories for rockmate
Users that are interested in rockmate are comparing it to the libraries listed below
Sorting:
- ☆220Updated 2 years ago
- ☆10Updated 3 years ago
- ☆159Updated 2 years ago
- Collection of kernels written in Triton language☆167Updated 7 months ago
- ☆113Updated last year
- Fast Hadamard transform in CUDA, with a PyTorch interface☆259Updated last month
- Memory Optimizations for Deep Learning (ICML 2023)☆110Updated last year
- This repository contains the experimental PyTorch native float8 training UX☆225Updated last year
- pytorch-profiler☆51Updated 2 years ago
- extensible collectives library in triton☆91Updated 7 months ago
- Distributed K-FAC preconditioner for PyTorch☆91Updated this week
- ☆182Updated last year
- A library for unit scaling in PyTorch☆132Updated 4 months ago
- Block-sparse primitives for PyTorch☆160Updated 4 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆137Updated 3 years ago
- Autonomous GPU Kernel Generation via Deep Agents☆137Updated this week
- A bunch of kernels that might make stuff slower 😉☆64Updated this week
- ☆43Updated 3 years ago
- Official PyTorch Implementation of HELP: Hardware-adaptive Efficient Latency Prediction for NAS via Meta-Learning (NeurIPS 2021 Spotlight…☆63Updated last year
- Butterfly matrix multiplication in PyTorch☆176Updated 2 years ago
- ☆60Updated last year
- Research and development for optimizing transformers☆131Updated 4 years ago
- PyTorch extension for emulating FP8 data formats on standard FP32 Xeon/GPU hardware.☆112Updated 11 months ago
- Experiment of using Tangent to autodiff triton☆80Updated last year
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆286Updated this week
- ☆37Updated last year
- ☆94Updated last year
- ☆43Updated last year
- ☆31Updated last year
- Personal solutions to the Triton Puzzles☆20Updated last year