mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/interpreters
History
George Necula 428377afb3
Added type annotations and removed unused imports (#2472) 
* Added type annotations and removed unused imports

* Adjusted type hints for pytype
2020-03-21 13:54:30 +01:00
..
__init__.py
populating source tree
2018-11-17 18:03:33 -08:00
ad.py
Add mypy type checking (#2430)
2020-03-18 17:06:05 -04:00
batching.py
Added type annotations and removed unused imports (#2472)
2020-03-21 13:54:30 +01:00
masking.py
Add mypy type checking (#2430)
2020-03-18 17:06:05 -04:00
parallel.py
Added type annotations and removed unused imports (#2472)
2020-03-21 13:54:30 +01:00
partial_eval.py
Added type annotations and removed unused imports (#2472)
2020-03-21 13:54:30 +01:00
pxla.py
Added type annotations and removed unused imports (#2472)
2020-03-21 13:54:30 +01:00
sharded_jit.py
Add mypy type checking (#2430)
2020-03-18 17:06:05 -04:00
xla.py
Added type annotations and removed unused imports (#2472)
2020-03-21 13:54:30 +01:00