mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-23 22:26:04 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/state
History
Peter Hawkins 319ab98980 Apply pyupgrade --py39-plus. 
Notable changes:
* use PEP 585 type names
* use PEP 604 type union syntax where `from __future__ import annotations` is present.
* use f-strings in more places.
* remove redundant arguments to open().
2023-07-21 14:49:44 -04:00
..
__init__.py
Add initial implementation of a run_state primitive
2023-04-03 21:32:32 -07:00
discharge.py
Apply pyupgrade --py39-plus.
2023-07-21 14:49:44 -04:00
primitives.py
Use lower-case PEP 585 names for types.
2023-06-23 15:12:14 -07:00
types.py
Apply pyupgrade --py39-plus.
2023-07-21 14:49:44 -04:00
utils.py
Add initial implementation of a run_state primitive
2023-04-03 21:32:32 -07:00