mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 19:56:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/state
History
Peter Hawkins 6cc1bf54a1 Move jax.interpreters.partial_eval to jax._src.interpreters.partial_eval. 
Also fix up some other internal imports of jax.interpreters.* to use jax._src.interpreters.

PiperOrigin-RevId: 519813664
2023-03-27 13:30:47 -07:00
..
__init__.py
Refactor Ref abstract type to contain other AbstractValues
2023-02-23 17:02:40 -08:00
discharge.py
Move jax.interpreters.partial_eval to jax._src.interpreters.partial_eval.
2023-03-27 13:30:47 -07:00
primitives.py
Move jax.interpreters.partial_eval to jax._src.interpreters.partial_eval.
2023-03-27 13:30:47 -07:00
types.py
Replace jax._src.util.prod with math.prod.
2023-02-28 12:41:00 -08:00