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

注意

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