mirrors/rocm_jax
1
0
Fork 0
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
Yash Katariya b38e85b3a4 Package utils.cc properly in jaxlib so that if jaxlib nightly is installed and then used, jaxlib_utils can be accessed. 
PiperOrigin-RevId: 523374835
2023-04-11 05:38:30 -07:00
..
rocm
[ROCm]: Run pmap test on specific number of GPUs
2023-03-30 18:34:47 +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
Package utils.cc properly in jaxlib so that if jaxlib nightly is installed and then used, jaxlib_utils can be accessed.
2023-04-11 05:38:30 -07:00
BUILD.bazel
Delete remote TPU support.
2023-03-24 12:33:33 -07:00
build.py
Merge pull request #15205 from yhtang:editable-jaxlib-build
2023-03-27 06:33:31 -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