mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/build
History
Rahul Batra ef79c19093 [ROCm]: Dockerfile and build script updates 
Add hipblaslt in Dockerfile
	Update docker file to default to ROCm5.6
 	CI scripts update to handle multiple ROCm versions
2023-09-08 22:56:59 +00:00
..
rocm
[ROCm]: Dockerfile and build script updates
2023-09-08 22:56:59 +00:00
build.py
Check build and wheel are installed before building jaxlib.
2023-07-26 11:46:11 -07:00
collect-profile-requirements.txt
Cloud TPU CI: make sure we update test deps and upgrade protobuf version
2023-07-19 16:17:47 -07:00
parallel_accelerator_execute.sh
Add script for parallel accelerator testing under Bazel.
2022-07-06 10:58:04 -04:00
test-requirements.txt
[Pallas] Upstream pallas to JAX
2023-08-01 16:43:13 -07:00