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/jaxlib
History
Skye Wanderman-Milne fc39332c4f
Clarify when jaxlib version should be bumped. (#4250)
2020-09-10 15:53:04 -07:00
..
BUILD
Use pytree defined in tensorflow. (#4087)
2020-08-18 08:58:43 +03:00
cublas.cc
Fix test failure on TPU. (#2088)
2020-01-27 12:48:10 -05:00
cuda_prng_kernels.cc
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
cuda_prng_kernels.cu.cc
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
cuda_prng_kernels.h
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
cuda_prng.py
Fix remaining flakes and use exclude within setup.cfg (#3371)
2020-06-08 22:58:03 -07:00
cusolver.cc
Fix test failure on TPU. (#2088)
2020-01-27 12:48:10 -05:00
cusolver.py
Update JAX to avoid XLA:Python API names deprecated in jaxlib 0.1.46. (#3046)
2020-05-11 17:43:55 -04:00
gpu_kernel_helpers.cc
Implement batched cholesky decomposition using LAPACK/Cusolver (#1956)
2020-01-07 10:56:15 -05:00
gpu_kernel_helpers.h
Implement batched cholesky decomposition using LAPACK/Cusolver (#1956)
2020-01-07 10:56:15 -05:00
kernel_helpers.h
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
kernel_pybind11_helpers.h
Fix jaxlib build by not exposing nvcc to pybind11. (#1819)
2019-12-05 18:59:29 -08:00
lapack.pyx
Parameterize geev/eig to allow for not computing the left/right eigenvectors (#3882)
2020-08-05 13:58:30 +03:00
version.py
Clarify when jaxlib version should be bumped. (#4250)
2020-09-10 15:53:04 -07:00