libffcv / ffcv
FFCV: Fast Forward Computer Vision (and other ML workloads!)
☆2,880Updated 7 months ago
Alternatives and similar repositories for ffcv:
Users that are interested in ffcv are comparing it to the libraries listed below
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,264Updated 10 months ago
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆2,426Updated last month
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,177Updated this week
- Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.☆1,823Updated 3 weeks ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,067Updated 9 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,655Updated last week
- functorch is JAX-like composable function transforms for PyTorch.☆1,403Updated this week
- solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning☆1,458Updated 3 weeks ago
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,704Updated last week
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,057Updated last month
- View model summaries in PyTorch!☆2,662Updated this week
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,412Updated 5 months ago
- Tensors, for human consumption☆1,178Updated last month
- Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.☆2,516Updated 3 weeks ago
- PyTorch extensions for high performance and large scale training.☆3,232Updated this week
- A python library for self-supervised learning on images.☆3,249Updated this week
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,080Updated last month
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆4,985Updated last week
- Official DeiT repository☆4,114Updated 10 months ago
- PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO☆6,525Updated 6 months ago
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,807Updated 10 months ago
- Model interpretability and understanding for PyTorch☆5,033Updated this week
- Scenic: A Jax Library for Computer Vision Research and Beyond☆3,385Updated this week
- Python 3.8+ toolbox for submitting jobs to Slurm☆1,355Updated 3 months ago
- 🐍 Geometric Computer Vision Library for Spatial AI☆10,143Updated this week
- Schedule-Free Optimization in PyTorch☆2,061Updated last month
- TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.☆1,519Updated this week
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,066Updated last month
- PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882☆2,023Updated last year