NVIDIA / apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
☆8,650Updated this week
Alternatives and similar repositories for apex
Users that are interested in apex are comparing it to the libraries listed below
Sorting:
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,935Updated 2 weeks ago
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,392Updated this week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,903Updated 2 weeks ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,040Updated last year
- Count the MACs / FLOPs of your PyTorch model.☆4,985Updated 10 months ago
- Datasets, Transforms and Models specific to Computer Vision☆16,824Updated this week
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆8,693Updated this week
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆14,908Updated last week
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,082Updated 3 years ago
- 🐍 Geometric Computer Vision Library for Spatial AI☆10,443Updated this week
- A PyTorch implementation of EfficientNet☆8,098Updated 3 years ago
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,135Updated 3 weeks ago
- ☆11,336Updated 2 months ago
- OpenMMLab Computer Vision Foundation☆6,116Updated 2 weeks ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆23,043Updated this week
- Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.☆29,439Updated this week
- An easy to use PyTorch to TensorRT converter☆4,732Updated 8 months ago
- PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722☆4,971Updated 2 weeks ago
- Model interpretability and understanding for PyTorch☆5,209Updated this week
- A small package to create visualizations of PyTorch execution graphs☆3,352Updated 4 months ago
- PyTorch extensions for high performance and large scale training.☆3,313Updated 2 weeks ago
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,472Updated 3 weeks ago
- Visualizer for neural network, deep learning and machine learning models☆30,171Updated this week
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,126Updated 5 months ago
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,745Updated this week
- A curated list of awesome self-supervised methods☆6,273Updated 10 months ago
- PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.☆6,925Updated 5 months ago
- An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…☆14,186Updated 10 months ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,109Updated last year
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆34,105Updated this week