This website requires JavaScript.
Explore
Help
Sign In
mirrors
/
rocm_jax
Watch
1
Star
0
Fork
0
You've already forked rocm_jax
mirror of
https://github.com/ROCm/jax.git
synced
2025-04-18 04:46:06 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
_src
/
interpreters
History
Yash Katariya
a9e48af260
Deprecated xla_call_p since it has been replaced with pjit.pjit_p
...
PiperOrigin-RevId: 518921538
2023-03-23 11:44:42 -07:00
..
__init__.py
Reapply: move
jax.interpreters.ad
to
jax._src.interpreters.ad
2023-02-02 09:29:05 -08:00
ad.py
Revert:
custom_vjp
symbolic zeros support
2023-03-22 09:56:09 -07:00
batching.py
Revert:
custom_vjp
symbolic zeros support
2023-03-22 09:56:09 -07:00
mlir.py
Deprecated xla_call_p since it has been replaced with pjit.pjit_p
2023-03-23 11:44:42 -07:00
pxla.py
Deprecated xla_call_p since it has been replaced with pjit.pjit_p
2023-03-23 11:44:42 -07:00
xla.py
Deprecated xla_call_p since it has been replaced with pjit.pjit_p
2023-03-23 11:44:42 -07:00
