mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 07:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/extend
History
Dan Foreman-Mackey 61701af4a2 Rename vmap methods for callbacks.
2024-10-21 15:03:04 -04:00
..
__init__.py
add jax.extend.random.wrap_key_data
2023-08-26 11:39:25 -07:00
ffi.py
Rename vmap methods for callbacks.
2024-10-21 15:03:04 -04:00
random.py
Run pyupgrade --py310-plus.
2024-06-26 16:10:18 -04:00