JAX
加拿大
代码工具模型工具

JAX

JAX: High-Performance Array Computing — JAX documentation