Sonnet是DeepMind为其旗舰深度学习框架JAX设计的高级神经网络库，如果你了解TensorFlow，可以把Sonnet类比为TensorFlow的KerasAPI，但它与JAX的函数式编程范式结合得更加...
