pytorch / torchcodecLinks
PyTorch media decoding and encoding
☆679Updated last 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☆295Updated this week
- Helpful tools and examples for working with flex-attention☆951Updated last week
- Transform datasets at scale. Optimize datasets for fast AI model training.☆529Updated last week
- TensorDict is a pytorch dedicated tensor container.☆955Updated last week
- For optimization algorithm research and development.☆530Updated this week
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆427Updated 8 months ago
- PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily wri…☆1,395Updated this week
- A suite of image and video neural tokenizers☆1,669Updated 6 months ago
- PyTorch native quantization and sparsity for training and inference☆2,291Updated this week
- A lightweight, local-first, and free experiment tracking library from Hugging Face 🤗☆710Updated this week
- State-of-the-art Image & Video CLIP, Multimodal Large Language Models, and More!☆1,542Updated last week
- UNet diffusion model in pure CUDA☆649Updated last year
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆389Updated 2 weeks ago
- A pytorch quantization backend for optimum☆984Updated this week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆571Updated 2 weeks ago
- Muon is an optimizer for hidden layers in neural networks☆1,595Updated last month
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆400Updated 4 months ago
- Best practices & guides on how to write distributed pytorch training code☆470Updated 6 months ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆116Updated 4 months ago
- Pytorch to Keras/Tensorflow/TFLite conversion made intuitive☆322Updated 5 months ago
- ☆516Updated last month
- Simple and readable code for training and sampling from diffusion models☆594Updated 2 months ago
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,510Updated 7 months ago
- ☆307Updated last year
- This repository provides the code and model checkpoints for AIMv1 and AIMv2 research projects.☆1,359Updated 3 weeks ago
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆120Updated 6 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆292Updated 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…☆236Updated last week
- Schedule-Free Optimization in PyTorch☆2,202Updated 3 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆284Updated last month