mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 10:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/extend
History
Dan Foreman-Mackey 1e206880d3 Move jax.ffi submodule to jax.extend.ffi
2024-05-31 12:34:59 -04:00
..
__init__.py
add jax.extend.random.wrap_key_data
2023-08-26 11:39:25 -07:00
ffi.py
Move jax.ffi submodule to jax.extend.ffi
2024-05-31 12:34:59 -04:00
random.py
Upgrade most .py sources to 3.9
2023-12-08 12:23:15 +00:00