pytorch / torchcodec
PyTorch video decoding
☆256Updated this week
Alternatives and similar repositories for torchcodec:
Users that are interested in torchcodec are comparing it to the libraries listed below
- Scalable and Performant Data Loading☆224Updated this week
- Transform datasets at scale. Optimize datasets for fast AI model training.☆422Updated this week
- PyTorch per step fault tolerance (actively under development)☆265Updated this week
- Helpful tools and examples for working with flex-attention☆679Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆222Updated 7 months ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆525Updated 3 weeks ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆359Updated last month
- ☆301Updated 8 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆120Updated 7 months ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆227Updated last week
- When it comes to optimizers, it's always better to be safe than sorry☆213Updated 2 weeks ago
- TensorDict is a pytorch dedicated tensor container.☆891Updated this week
- ☆377Updated last week
- Muon optimizer: +>30% sample efficiency with <3% wallclock overhead☆494Updated this week
- TorchFix - a linter for PyTorch-using code with autofix support☆136Updated last month
- Megatron's multi-modal data loader☆171Updated this week
- ☆86Updated last year
- ☆156Updated last month
- Optimized library for large-scale extraction of frames and audio from video.☆203Updated last year
- For optimization algorithm research and development.☆498Updated 2 weeks ago
- ☆148Updated last year
- Efficient LLM Inference over Long Sequences☆362Updated last month
- FlashRNN - Fast RNN Kernels with I/O Awareness☆76Updated this week
- Inference Vision Transformer (ViT) in plain C/C++ with ggml☆260Updated 11 months ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆228Updated last month
- Official Implementation for our NeurIPS 2024 paper, "Don't Look Twice: Run-Length Tokenization for Faster Video Transformers".☆199Updated 3 months ago
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆506Updated 4 months ago
- Best practices & guides on how to write distributed pytorch training code☆367Updated 2 weeks ago
- An efficient video loader for deep learning with smart shuffling that's super easy to digest☆31Updated last year