mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
3 Commits 500 Branches 263 Tags
Go to file
Clone
Open with VS Code Open with VSCodium Open with Intellij IDEA
Matthew Johnson f78c05195d add gpu build script
2018-11-18 14:24:09 -08:00
build
add gpu build script
2018-11-18 14:24:09 -08:00
jax
change xla_data_pb2 import
2018-11-17 22:58:30 -08:00
.gitignore
add gpu build script
2018-11-18 14:24:09 -08:00
WORKSPACE
populating source tree
2018-11-17 18:03:33 -08:00
Description
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Apache-2.0 146 MiB
Languages
Python 89.3%
C++ 8.1%
Starlark 1.5%
Jupyter Notebook 0.5%
Shell 0.3%
Other 0.3%