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
Rahul Batra b0e541a730 [ROCm]: Updates for container and build script 
-Updated dockerfile.ms
	-Updated build script to switch building against XLA repo
  	-Update CI script
	-Update jaxlib setup.py to add rocm version
2023-06-19 18:13:28 +00:00
..
rocm
[ROCm]: Updates for container and build script
2023-06-19 18:13:28 +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
Move triton_kernel_call_lib to jaxlib
2023-05-24 12:11:21 -07:00
BUILD.bazel
build/BUILD.bazel: remove unused import
2023-06-12 18:36:32 -04:00
build.py
Update bazel version to 6.1.2.
2023-05-10 10:57:29 -04: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
Add NumPy as a test requirement.
2023-06-14 10:14:56 -04:00