Lightning-Universe / lightning-flash
Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
☆1,743Updated last year
Related projects ⓘ
Alternatives and complementary repositories for lightning-flash
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,691Updated this week
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,134Updated this week
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,131Updated this week
- Generic template to bootstrap your PyTorch project.☆635Updated last year
- MADGRAD Optimization Method☆801Updated last year
- Pytorch Lightning code guideline for conferences☆1,239Updated last year
- An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come☆848Updated this week
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,257Updated 8 months ago
- Flexible components pairing 🤗 Transformers with Pytorch Lightning☆610Updated last year
- Version control for machine learning☆1,650Updated 2 months ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,400Updated 3 months ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,038Updated 7 months ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,523Updated this week
- The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflo…☆794Updated 2 years ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,857Updated 4 months ago
- PyTorch extensions for high performance and large scale training.☆3,187Updated 2 months ago
- Model interpretability and understanding for PyTorch☆4,918Updated this week
- A python library for self-supervised learning on images.☆3,157Updated this week
- Hummingbird compiles trained ML models into tensor computation for faster inference.☆3,352Updated 2 weeks ago
- Over 200 figures and diagrams of the most popular deep learning architectures and layers FREE TO USE in your blog posts, slides, presenta…☆1,387Updated 3 years ago
- TensorFlow Similarity is a python package focused on making similarity learning quick and easy.☆1,013Updated 6 months ago
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,470Updated last month
- Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models☆2,064Updated 3 months ago
- Accelerated deep learning R&D☆3,291Updated 7 months ago
- Automatic architecture search and hyperparameter optimization for PyTorch☆2,371Updated 7 months ago
- ML Collections is a library of Python Collections designed for ML use cases.☆893Updated 3 months ago
- Experiment tracking, ML developer tools☆870Updated last year
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆2,303Updated last month
- Library for 8-bit optimizers and quantization routines.