Jax的主要应用场景：
- 深度学习 (Deep Learning)：JAX 在深度学习场景下应用很广泛，很多团队基于 JAX 开发了更加高级的 API 支持不同的场景，方便开发者使用。
- 科学模拟 (Scientific Simulation)：JAX 的出现不仅仅是针对于深度学习，其实也拥有很多其他的使命，如科学模拟。
- 机器人与控制系统 (Robotics and Control Systems)
- 概率编程 (Probabilistic Programming)
Jax的升级包：
深度学习框架（Google）：
https://github.com/google-deepmind/optax