Massively parallel rigidbody physics simulation on accelerator hardware.
created at June 2, 2021, 9:45 p.m.
Fast and Easy Infinite Neural Networks in Python
created at April 8, 2019, 4:48 p.m.
Flax is a neural network library for JAX that is designed for flexibility.
created at Jan. 10, 2020, 9:48 a.m.
Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.
created at July 26, 2018, 9:58 a.m.