mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 10:06:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/state
History
Peter Hawkins 8fb1fd318d Replace jax._src.util.prod with math.prod. 
math.prod() was added in Python 3.8, so we can assume it is always present.

PiperOrigin-RevId: 513011144
2023-02-28 12:41:00 -08:00
..
__init__.py
Refactor Ref abstract type to contain other AbstractValues
2023-02-23 17:02:40 -08:00
discharge.py
Refactor Ref abstract type to contain other AbstractValues
2023-02-23 17:02:40 -08:00
primitives.py
Refactor Ref abstract type to contain other AbstractValues
2023-02-23 17:02:40 -08:00
types.py
Replace jax._src.util.prod with math.prod.
2023-02-28 12:41:00 -08:00