jax
16bd2725 - use lax instead of numpy to avoid circular import

Commit

6 years ago

use lax instead of numpy to avoid circular import

References

#1268 - Fix cond batching rule issues

Author

fehiepsi

fehiepsi

Parents

Loading