Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
Repository Index
47,133 repositories indexed
1–4 of 4 · Most stars
Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...
- reinforcement-learning
- deep-learning
- pytorch
- pytorch-reinforcement-learning
An Easy To Use PyTorch Computer Vision Library
- computer-vision
- deep-learning
- pytorch
- pytorch-lightning
A minimalistic codebase for training NLG models from HuggingFace Transformers using PyTorch Lightning.
- bart
- generation
- gpt-2
- language-model