Massively parallel rigidbody physics simulation
VMAS is a vectorized differentiable simulator
Language Model Reinforcement Learning Environments frameworks
Monte Carlo tree search in JAX
A unified framework for scalable computing
A collection of reference Jupyter notebooks and demo AI/ML application
Massively Parallel Deep Reinforcement Learning
Reinforced Recommendation toolkit built around pytorch 1.7
A Neural Net Training Interface on TensorFlow, with focus on speed
Intel® Nervana™ reference deep learning framework