mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
Add type annotations to user-facing functions in lax.py (#2644)
* Add type annotations to user-facing functions in lax.py * Remove redundant comment.
This commit is contained in:
parent
fcefc7e8f1
commit
5174f6d5dc
530
jax/lax/lax.py
530
jax/lax/lax.py
File diff suppressed because it is too large
Load Diff
Loading…
x
Reference in New Issue
Block a user