facebookresearch / dietgpuLinks
GPU implementation of a fast generalized ANS (asymmetric numeral system) entropy encoder and decoder, with extensions for lossless compression of numerical and other data types in HPC/ML applications.
☆338Updated last week
Alternatives and similar repositories for dietgpu
Users that are interested in dietgpu are comparing it to the libraries listed below
Sorting:
- A library to analyze PyTorch traces.☆391Updated this week
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆235Updated 2 years ago
- A tensor-aware point-to-point communication primitive for machine learning☆258Updated 2 years ago
- Implementation of a Transformer, but completely in Triton☆269Updated 3 years ago
- ☆318Updated last week
- An open-source efficient deep learning framework/compiler, written in python.☆704Updated last week
- A GPU performance profiling tool for PyTorch models☆503Updated 3 years ago
- Fast Block Sparse Matrices for Pytorch☆548Updated 4 years ago
- Named tensors with first-class dimensions for PyTorch☆332Updated 2 years ago
- jax-triton contains integrations between JAX and OpenAI Triton☆403Updated this week
- A library of GPU kernels for sparse matrix operations.☆265Updated 4 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆132Updated 3 years ago
- This is a Tensor Train based compression library to compress sparse embedding tables used in large-scale machine learning models such as …☆194Updated 2 years ago
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆180Updated 2 weeks ago
- Research and development for optimizing transformers☆129Updated 4 years ago
- ☆250Updated 11 months ago
- Customized matrix multiplication kernels☆56Updated 3 years ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 10 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆276Updated 3 weeks ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,053Updated last year
- MONeT framework for reducing memory consumption of DNN training☆173Updated 4 years ago
- MLIR-based partitioning system☆97Updated this week
- Library for 8-bit optimizers and quantization routines.☆715Updated 2 years ago
- A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.☆821Updated this week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆170Updated this week
- Block-sparse primitives for PyTorch☆156Updated 4 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- Accelerate PyTorch models with ONNX Runtime☆362Updated 4 months ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- A collection of tools for neural compression enthusiasts.☆561Updated 9 months ago