libffcv / ffcv
FFCV: Fast Forward Computer Vision (and other ML workloads!)
☆2,831Updated 3 months ago
Related projects: ⓘ
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,249Updated 6 months ago
- A python library for self-supervised learning on images.☆2,923Updated this week
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆2,206Updated this week
- Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.☆2,077Updated this week
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,678Updated this week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,362Updated this week
- Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.☆1,816Updated 3 weeks ago
- PyTorch extensions for high performance and large scale training.☆3,149Updated 2 weeks ago
- Tensors, for human consumption☆1,096Updated 2 weeks ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,012Updated 5 months ago
- solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning☆1,411Updated last month
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,002Updated 3 months ago
- Scenic: A Jax Library for Computer Vision Research and Beyond☆3,248Updated 3 weeks ago
- Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.☆2,206Updated 3 weeks ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆1,981Updated last month
- Model interpretability and understanding for PyTorch☆4,810Updated this week
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,382Updated last month
- View model summaries in PyTorch!☆2,499Updated last week
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆4,573Updated last week
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,116Updated this week
- Pytorch Lightning code guideline for conferences☆1,228Updated 11 months ago
- Official DeiT repository☆4,005Updated 6 months ago
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,742Updated 6 months ago
- Code release for ConvNeXt model☆5,703Updated last year
- Code for visualizing the loss landscape of neural nets☆2,770Updated 2 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,389Updated this week
- Geometric Computer Vision Library for Spatial AI☆9,806Updated this week
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆5,949Updated last week
- A smaller subset of 10 easily classified classes from Imagenet, and a little more French☆956Updated last year
- Python 3.8+ toolbox for submitting jobs to Slurm☆1,255Updated 2 weeks ago