Rohith Gandhi G's starred repositories
albumentations
Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
pytorch-metric-learning
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
segmentation_models
Segmentation models with pretrained backbones. Keras and TensorFlow Keras.
pytorch-a2c-ppo-acktr-gail
PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).
hummingbird
Hummingbird compiles trained ML models into tensor computation for faster inference.
pytorchvideo
A deep learning library for video understanding research.
pytorch-optimizer
torch-optimizer -- collection of optimizers for Pytorch
knockknock
🚪✊Knock Knock: Get notified when your training ends with only two additional lines of code
CompilerGym
Reinforcement learning environments for compiler and program optimization tasks
3dgan-release
3D Generative Adversarial Network
slimevolleygym
A simple OpenAI Gym environment for single and multi-agent reinforcement learning
pytorch-trpo
PyTorch implementation of Trust Region Policy Optimization
futurepast
Deprecation tools for Python
tamil-nlp-catalog
Awesome List of Tamil NLP & AI Resources