pytorch / nestedtensor
[Prototype] Tools for the concurrent manipulation of variably sized Tensors.
☆253Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for nestedtensor
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated last year
- Fast Block Sparse Matrices for Pytorch☆545Updated 3 years ago
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆205Updated 2 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆328Updated 2 years ago
- Implementation of a Transformer, but completely in Triton☆249Updated 2 years ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆216Updated last week
- Slicing a PyTorch Tensor Into Parallel Shards☆296Updated 3 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,396Updated this week
- Block-sparse primitives for PyTorch☆148Updated 3 years ago
- ☆155Updated 4 years ago
- Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆428Updated 2 months ago
- MONeT framework for reducing memory consumption of DNN training☆173Updated 3 years ago
- ☆328Updated this week
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆124Updated 2 years ago
- A library to inspect and extract intermediate layers of PyTorch models.☆470Updated 2 years ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆473Updated 2 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆561Updated this week
- A GPU performance profiling tool for PyTorch models☆495Updated 3 years ago
- Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…☆294Updated this week
- A Pytree Module system for Deep Learning in JAX☆214Updated last year
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆180Updated 3 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆179Updated last year
- JMP is a Mixed Precision library for JAX.☆187Updated 6 months ago
- A collection of code snippets for my PyTorch Lightning projects☆107Updated 3 years ago
- PyTorch RFCs (experimental)☆130Updated 2 months ago
- PyTorch implementation of L2L execution algorithm☆106Updated last year
- Accelerate PyTorch models with ONNX Runtime☆356Updated 2 months ago
- Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)☆262Updated 4 years ago
- jax-triton contains integrations between JAX and OpenAI Triton☆343Updated 3 weeks ago