This website requires JavaScript.
Explore
Help
Sign In
mirrors
/
rocm_jax
Watch
1
Star
0
Fork
0
You've already forked rocm_jax
mirror of
https://github.com/ROCm/jax.git
synced
2025-04-17 20:36:05 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
build
History
Jieying Luo
0ce7c7b7bd
Register plugin profiler for TPU and remove
--config=tpu
/
--enable_tpu
in jaxlib.
...
PiperOrigin-RevId: 580561059
2023-11-08 09:40:28 -08:00
..
rocm
[ROCm]: Dockerfile and build script updates
2023-09-08 22:56:59 +00:00
build.py
Register plugin profiler for TPU and remove
--config=tpu
/
--enable_tpu
in jaxlib.
2023-11-08 09:40:28 -08: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
Small improvements to build/build.py
2023-09-26 21:55:45 -04:00
