mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 05:26:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/extend
History
Dan Foreman-Mackey 79c222eee6 Fix bug in ffi_lowering where custom layouts were ignored. 
PiperOrigin-RevId: 664795687
2024-08-19 07:20:06 -07:00
..
__init__.py
add jax.extend.random.wrap_key_data
2023-08-26 11:39:25 -07:00
ffi.py
Fix bug in ffi_lowering where custom layouts were ignored.
2024-08-19 07:20:06 -07:00
random.py
Run pyupgrade --py310-plus.
2024-06-26 16:10:18 -04:00