jax
156024fb - check that jaxpr types match in cond transpose using `core.typematch` rather than `eq`

Commit

5 years ago

check that jaxpr types match in cond transpose using `core.typematch` rather than `__eq__`

References

Author

froystig

Parents

jax 156024fb - check that jaxpr types match in cond transpose using `core.typematch` rather than `__eq__`