Add type annotations to user-facing functions in lax.py (#2644) · jax-ml/jax@5174f6d · GitHub

Commit

Add type annotations to user-facing functions in lax.py (#2644)

Browse files

* Add type annotations to user-facing functions in lax.py

* Remove redundant comment.

Loading branch information

hawkinsp authored Apr 8, 2020

1 parent fcefc7e commit 5174f6d