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
Peter Hawkins
c2d6fcc0e6
Split core.py and several files in an SCC with it into a separate Bazel build target.
...
PiperOrigin-RevId: 520192610
2023-03-28 18:31:13 -07:00
..
__init__.py
Reapply: move
jax.interpreters.ad
to
jax._src.interpreters.ad
2023-02-02 09:29:05 -08:00
ad.py
Move jax.interpreters.partial_eval to jax._src.interpreters.partial_eval.
2023-03-27 13:30:47 -07:00
batching.py
Move jax.interpreters.partial_eval to jax._src.interpreters.partial_eval.
2023-03-27 13:30:47 -07:00
mlir.py
Move jax.interpreters.partial_eval to jax._src.interpreters.partial_eval.
2023-03-27 13:30:47 -07:00
partial_eval.py
Move jax.interpreters.partial_eval to jax._src.interpreters.partial_eval.
2023-03-27 13:30:47 -07:00
pxla.py
Split core.py and several files in an SCC with it into a separate Bazel build target.
2023-03-28 18:31:13 -07:00
xla.py
Split core.py and several files in an SCC with it into a separate Bazel build target.
2023-03-28 18:31:13 -07:00
