示例：编写一个简单的神经网络

注意

这是新的 JAX 教程草案 中一个章节的占位符。