文件名称:JAX是Autograd和XLA的结合,用于高性能机器学习研究。-Python开发
文件大小:9.62MB
文件格式:ZIP
更新时间:2024-06-14 04:39:32
Python Deep Learning
Python + NumPy程序的可组合转换:区分,矢量化,JIT到GPU / TPU,以及更多JAX:Autograd和XLA 转型| 安装指南| 神经网络库| 变更记录| 参考文档| 代码搜索新闻:JAX突破了最大规模的MLPerf Training 0.7基准测试! 什么是JAX? JAX是Autograd和XLA的结合,用于高性能机器学习研究。 借助其Autograd的更新版本,JAX可以自动区分本机Python和NumPy函数。 它可以通过循环,分支,递归和闭包来区分,并且可以用d