Home / JAX / Neural Network Libraries > Objax Neural Network Libraries > Objax Has an object oriented design similar to PyTorch. Package 775 stars Archived GitHub Back to JAX