pytorch / torchcodecLinks
PyTorch media decoding and encoding
☆589Updated this week
Alternatives and similar repositories for torchcodec
Users that are interested in torchcodec are comparing it to the libraries listed below
Sorting:
- Scalable and Performant Data Loading☆278Updated this week
- Helpful tools and examples for working with flex-attention☆846Updated this week
- TensorDict is a pytorch dedicated tensor container.☆929Updated this week
- PyTorch per step fault tolerance (actively under development)☆329Updated this week
- Transform datasets at scale. Optimize datasets for fast AI model training.☆493Updated last week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆556Updated last week
- UNet diffusion model in pure CUDA☆608Updated 11 months ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆426Updated 6 months ago
- State-of-the-art Image & Video CLIP, Multimodal Large Language Models, and More!☆1,294Updated 3 weeks ago
- For optimization algorithm research and development.☆521Updated this week
- ☆476Updated this week
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆381Updated 2 months ago
- Muon: An optimizer for hidden layers in neural networks☆897Updated 2 weeks ago
- Implementation of Diffusion Transformer (DiT) in JAX☆278Updated last year
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆253Updated this week
- PyTorch native quantization and sparsity for training and inference☆2,125Updated this week
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆110Updated 2 months ago
- Official repository for "AM-RADIO: Reduce All Domains Into One"☆1,211Updated 2 weeks ago
- ☆159Updated last year
- A suite of image and video neural tokenizers☆1,636Updated 4 months ago
- ☆303Updated last year
- Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and …☆1,367Updated this week
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆399Updated 2 weeks ago
- When it comes to optimizers, it's always better to be safe than sorry☆241Updated 2 months ago
- FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores☆319Updated 5 months ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 10 months ago
- A pytorch quantization backend for optimum☆955Updated last week
- Simple and readable code for training and sampling from diffusion models☆509Updated last week
- An implementation of PSGD Kron second-order optimizer for PyTorch☆91Updated 2 months ago
- PyTorch Single Controller☆231Updated this week