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-24 13:56:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
interpreters
History
jax authors
ee0292c6d2
Merge pull request
#24470
from jakevdp:is-unspecified
...
PiperOrigin-RevId: 689820689
2024-10-25 09:49:30 -07:00
..
__init__.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
ad.py
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
batching.py
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
mlir.py
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
partial_eval.py
Remove some untested dynamic shapes paths (prep work for stackless).
2024-09-19 12:59:48 -07:00
pxla.py
sharding cleanup: use inline checks for unimplemented and auto
2024-10-25 04:22:40 -07:00
xla.py
Finalize deprecation of xb, xc, & xe symbols in jax.interpreters.xla
2024-10-25 08:12:44 -07:00