Added Numba accelerated PyTorch optimizers

rapidsai / deeplearning

337 stars 90 forks source link

Closed madsbk closed 5 years ago

madsbk commented 5 years ago

This PR adds Numba accelerated optimizers: adam.py, adamw.py, and radam.py.