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-16 11:56:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
build
History
Jieying Luo
16b4f69769
Rename arg in build script to be more clear.
...
The flag means skips GPU plugin extension in jaxlib. PiperOrigin-RevId: 627203738
2024-04-22 17:22:24 -07:00
..
rocm
[ROCM]: Generating pytest html logs from unit-tests.
2024-01-24 15:08:35 +00:00
build.py
Rename arg in build script to be more clear.
2024-04-22 17:22:24 -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
Evaluate the correctness of JAX complex functions using mpmath as a reference
2024-03-21 23:35:29 +02:00