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 07:46:06 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
_src
/
export
History
Yash Katariya
6efcf44b1a
Deprecate
PositionalSharding
and
GSPMDSharding
...
PiperOrigin-RevId: 746564071
2025-04-11 13:06:43 -07:00
..
__init__.py
[export] Move the export implementation to jax._src.export.
2024-06-09 08:59:50 -07:00
_export.py
Deprecate
PositionalSharding
and
GSPMDSharding
2025-04-11 13:06:43 -07:00
serialization_generated.py
[export] Add support for serializing functions with PRNG keys as inputs/outputs
2025-04-07 11:53:20 +02:00
serialization.fbs
[export] Add support for serializing functions with PRNG keys as inputs/outputs
2025-04-07 11:53:20 +02:00
serialization.py
[export] Add support for serializing functions with PRNG keys as inputs/outputs
2025-04-07 11:53:20 +02:00
shape_poly_decision.py
[shape_poly] Improve threefry with symbolic shapes
2025-01-07 09:10:04 +02:00
shape_poly.py
Replace references to jax.readthedocs.io with docs.jax.dev.
2025-04-08 08:33:49 -07:00
