Google主打的机器学习计算框架——jax的升级包

相关：
机器学习洞察 | 一文带你“讲透” JAX

Jax的主要应用场景：

• 深度学习 (Deep Learning)：JAX 在深度学习场景下应用很广泛，很多团队基于 JAX 开发了更加高级的 API 支持不同的场景，方便开发者使用。
• 科学模拟 (Scientific Simulation)：JAX 的出现不仅仅是针对于深度学习，其实也拥有很多其他的使命，如科学模拟。
• 机器人与控制系统 (Robotics and Control Systems)
• 概率编程 (Probabilistic Programming)

---

Jax的升级包：
深度学习框架（Google）：
https://github.com/google-deepmind/optax