Implementation of One-Cycle Learning rate policy (adapted from Fast.ai lib)
MIT License
PyTorch Implementation of Soft Actor-Critic Algorithm
Reproduce the results of "Neuroevolution of Self-Interpretable Agents" paper
Implementation of "Perceptual Losses for Real-Time Style Transfer and Super-Resolution" in Keras
Time series forecasting with PyTorch
The PyTorch Implementation based on YOLOv4 of the paper: "Complex-YOLO: Real-time 3D Object Detec...
ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning mod...
Yet Another Neural Network Library 🤔
A minimal reproduction of LCZero's training code, for ease of experimentation and benchmarking
An implementation of training for GPT2, supports TPUs
Message Passing Neural Networks for Molecule Property Prediction
Schedule-Free Optimization in PyTorch