mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/build
History
Peter Hawkins b943b31b22
Add jax.image.resize. (#3703) 
* Add jax.image.resize.

This is a port of `tf.image.resize()` and the `ScaleAndTranslate` operator.

While I don't expect this implementation to be particularly fast, it is a useful generic implementation to which we can add optimized special cases as the need arises.
2020-07-10 09:57:59 -04:00
..
jaxlib
Remove old xla_data_pb2 compatibility shim.
2019-07-29 15:21:47 -04:00
build_jaxlib_wheels_macos.sh
Minor improvements to the script to build macos wheels (#3013)
2020-05-11 20:17:26 +03:00
build_jaxlib_wheels.sh
Make CUDA wheels manylinux2010 compliant, add CUDA 11, drop CUDA 9.2 (#3555)
2020-06-25 14:37:14 -04:00
build_wheel_docker_entrypoint.sh
Make CUDA wheels manylinux2010 compliant, add CUDA 11, drop CUDA 9.2 (#3555)
2020-06-25 14:37:14 -04:00
BUILD.bazel
Add a handwritten ThreeFry2x32 CUDA kernel. (#1756)
2019-11-24 13:06:23 -05:00
build.py
Make CUDA wheels manylinux2010 compliant, add CUDA 11, drop CUDA 9.2 (#3555)
2020-06-25 14:37:14 -04:00
Dockerfile
Drop support for Python 3.5. (#2445)
2020-03-18 10:54:28 -04:00
install_cuda.sh
Update docker script for CUDA 11. (#3571)
2020-06-26 11:20:51 -04:00
install_xla_in_source_tree.sh
Add a handwritten ThreeFry2x32 CUDA kernel. (#1756)
2019-11-24 13:06:23 -05:00
setup.py
Drop support for Python 3.5. (#2445)
2020-03-18 10:54:28 -04:00
test-requirements.txt
Add jax.image.resize. (#3703)
2020-07-10 09:57:59 -04:00