jax
73b76e99 - Exported lax from jax/init.py (#3135)

Commit

6 years ago

Exported lax from jax/__init__.py (#3135) This allows to use lax functions without separately importing jax.lax.

References

Author

superbobry

Parents

jax 73b76e99 - Exported lax from jax/__init__.py (#3135)