GitHub repository: mrhashemi/Neural_Net_JAX_Optax_Keras
Primary language: Jupyter Notebook
Created: 2025-01-02T13:25:22Z
Description: A simple neural network is created using JAX with optax optimizer and a custom defined loss function.