https://github.com/mrhashemi/neural_net_jax_optax_keras
A simple neural network is created using JAX with optax optimizer and a custom defined loss function.
https://github.com/mrhashemi/neural_net_jax_optax_keras
jax keras neural-network optax
Last synced: 2 months ago
JSON representation
A simple neural network is created using JAX with optax optimizer and a custom defined loss function.
- Host: GitHub
- URL: https://github.com/mrhashemi/neural_net_jax_optax_keras
- Owner: mrhashemi
- License: mit
- Created: 2025-01-02T13:25:21.000Z (4 months ago)
- Default Branch: main
- Last Pushed: 2025-02-04T13:41:03.000Z (3 months ago)
- Last Synced: 2025-02-04T14:39:29.353Z (3 months ago)
- Topics: jax, keras, neural-network, optax
- Language: Jupyter Notebook
- Homepage:
- Size: 84 KB
- Stars: 0
- Watchers: 1
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- License: LICENSE