mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/build
History
Yash Katariya cf6b5097d0 Remove pytest_benchmark for test-requirements.txt and move the benchmark file which was using that package to use google_benchmark. 
PiperOrigin-RevId: 483736267
2022-10-25 11:59:32 -07:00
..
rocm
[ROCm] Fix ROCm dockerfile to remove 2020-resolver opt in.
2022-10-19 18:11:04 +00:00
build_jaxlib_wheels_helpers.sh
Remove -it flag from docker run command in jaxlib build script.
2021-04-14 09:44:42 -07:00
build_jaxlib_wheels_macos.sh
Drop support for NumPy 1.18.
2022-01-04 12:11:38 -08:00
build_jaxlib_wheels.sh
Drop Python 3.6 support.
2021-07-15 14:20:29 -04:00
build_wheel_docker_entrypoint.sh
Allow manylinux_2_12_x86_64 in auditwheel test. This is an alias by PEP600.
2021-05-07 18:22:28 -04:00
build_wheel.py
Merge CUDA and ROCM kernel code in jaxlib.
2022-10-25 07:23:34 -07:00
BUILD.bazel
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
build.py
Merge pull request #11961 from jakeh-gc:plugin_device
2022-09-23 07:29:17 -07:00
Dockerfile
Drop support for NumPy 1.18.
2022-01-04 12:11:38 -08:00
install_cuda.sh
Update cuDNN versions to highest available.
2021-03-09 19:07:00 +00:00
LICENSE.txt
Switch from pocketfft to ducc
2022-08-26 13:36:25 +00:00
parallel_accelerator_execute.sh
Add script for parallel accelerator testing under Bazel.
2022-07-06 10:58:04 -04:00
test-requirements.txt
Remove pytest_benchmark for test-requirements.txt and move the benchmark file which was using that package to use google_benchmark.
2022-10-25 11:59:32 -07:00