pytorch / torchcodecLinks
PyTorch media decoding and encoding
☆624Updated 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☆288Updated last week
- Helpful tools and examples for working with flex-attention☆876Updated this week
- TensorDict is a pytorch dedicated tensor container.☆942Updated this week
- For optimization algorithm research and development.☆521Updated this week
- Transform datasets at scale. Optimize datasets for fast AI model training.☆506Updated this week
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆426Updated 7 months ago
- State-of-the-art Image & Video CLIP, Multimodal Large Language Models, and More!☆1,431Updated this week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆560Updated this week
- A suite of image and video neural tokenizers☆1,645Updated 5 months ago
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆361Updated this week
- Muon is an optimizer for hidden layers in neural networks☆1,092Updated this week
- UNet diffusion model in pure CUDA☆610Updated last year
- PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily wri…☆1,375Updated this week
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆385Updated 3 months ago
- ☆489Updated last week
- Implementation of Diffusion Transformer (DiT) in JAX☆279Updated last year
- PyTorch native quantization and sparsity for training and inference☆2,168Updated this week
- Simple and readable code for training and sampling from diffusion models☆511Updated last month
- Best practices & guides on how to write distributed pytorch training code☆450Updated 4 months ago
- This repository provides the code and model checkpoints for AIMv1 and AIMv2 research projects.☆1,326Updated 2 months ago
- PyTorch Single Controller☆318Updated this week
- When it comes to optimizers, it's always better to be safe than sorry☆302Updated 3 months ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆112Updated 3 months ago
- Schedule-Free Optimization in PyTorch☆2,189Updated last month
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,495Updated 6 months ago
- Pytorch to Keras/Tensorflow/TFLite conversion made intuitive☆317Updated 4 months ago
- ☆304Updated last year
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆235Updated 6 months ago
- Hiera: A fast, powerful, and simple hierarchical vision transformer.☆1,003Updated last year
- A pytorch quantization backend for optimum☆963Updated last week