rocm_jax/docs/tutorials/simple-neural-network.md

Example: Writing a simple neural network

This is a placeholder for a section in the new {ref}`jax-tutorials`.