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
Peter Hawkins e60d5dd54c
Remove "from __future__" uses from JAX. (#2117) 
The future (Python 3) has arrived; no need to request it explicitly.
2020-01-29 12:29:03 -05:00
..
BUILD
Add missing BUILD dependency. (#2089)
2020-01-27 13:15:41 -05: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
Remove "from __future__" uses from JAX. (#2117)
2020-01-29 12:29:03 -05:00
cusolver.cc
Fix test failure on TPU. (#2088)
2020-01-27 12:48:10 -05:00
cusolver.py
Remove "from __future__" uses from JAX. (#2117)
2020-01-29 12:29:03 -05: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
added batching to LAPACK triangular_solve (#1985)
2020-01-14 11:18:47 -05:00
pytree.cc
Avoid out-of-bounds dereference for arity-0 nodes. (#1713)
2019-11-18 15:35:07 -05:00
version.py
Bump jaxlib version to 0.1.38 and update WORKSPACE.
2020-01-21 16:59:27 -08:00