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

注意

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