Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Stars
35.3k
Forks
3.5k
Watchers
35.3k
Open Issues
2.2k
Overall repository health assessment
No package.json found
This might not be a Node.js project
3.5k
commits
3.4k
commits
2.8k
commits
2.6k
commits
2.1k
commits
1.2k
commits
1.1k
commits
856
commits
690
commits
502
commits
Update XLA dependency to use revision http://github.com/openxla/xla/commit/5797c796dc2bcf1ace4b7852c8e0e393c0812317
54a9d3bView on GitHubMerge pull request #36502 from mattjj:get-transpose-fix
4435b64View on GitHub[jaxlib] Added all missing .pyis for jaxlib native extensions
52669c9View on GitHubFinalize deprecation of several `jax.core` APIs for JAX v0.10.0
4b46b1cView on GitHubMerge pull request #36492 from magaonka-amd:fix/rocm-ci-test-skips
b7e3951View on GitHubMerge pull request #36426 from ROCm:rocm-disable-gesdd-default
63356f7View on GitHubMerge pull request #36350 from ezhulenev:autotuning-client-1
deee816View on GitHub