Last released Nov 6, 2023
Objax is a machine learning framework that provides an Object Oriented layer for JAX.
Supported by