jax-ml / jax

Python+NumPy 程序的可组合转换：微分、向量化、JIT 到 GPU/TPU 等

仓库 Star

36,569总榜 #890

Fork

4,016

最近推送

3 天前更新

仓库创建

2018 年创建