topal-team / rockmateLinks
☆36Updated 9 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)☆107Updated last year
- ☆159Updated 2 years ago
- ☆214Updated 2 years ago
- Collection of kernels written in Triton language☆154Updated 5 months ago
- Distributed K-FAC preconditioner for PyTorch☆90Updated this week
- ☆29Updated 10 months ago
- A bunch of kernels that might make stuff slower 😉☆59Updated this week
- ☆111Updated last year
- A library for unit scaling in PyTorch☆130Updated 2 months ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆134Updated 3 years ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated last year
- Complete GPU residency for ML.☆43Updated last month
- extensible collectives library in triton☆87Updated 5 months ago
- Butterfly matrix multiplication in PyTorch☆174Updated last year
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆307Updated this week
- ☆43Updated last year
- pytorch-profiler☆51Updated 2 years ago
- Experiment of using Tangent to autodiff triton☆81Updated last year
- Official PyTorch Implementation of HELP: Hardware-adaptive Efficient Latency Prediction for NAS via Meta-Learning (NeurIPS 2021 Spotlight…☆63Updated last year
- Fast Hadamard transform in CUDA, with a PyTorch interface☆233Updated 2 weeks ago
- ☆175Updated last year
- A block oriented training approach for inference time optimization.☆34Updated last year
- Research and development for optimizing transformers☆130Updated 4 years ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆223Updated this week
- ☆41Updated last year
- PyTorch extension for emulating FP8 data formats on standard FP32 Xeon/GPU hardware.☆111Updated 9 months ago
- ☆33Updated last year
- PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021☆56Updated 4 years ago
- PyTorch bindings for CUTLASS grouped GEMM.☆119Updated 3 months ago