Neural Networks in Equinox (JAX DL framework) with Optax

Neural Networks in Equinox (JAX DL framework) with Optax

@MachineLearningSimulation, June 2023