jax
498ef4e7 - Change core.valid_jaxtype to allow tracers.

Commit
52 days ago
Change core.valid_jaxtype to allow tracers. Almost every caller of this method wanted to allow tracers anyway. This does change the behavior of the public function jax.core.valid_jaxtype, but we don't really consider that a stable API in the first place. PiperOrigin-RevId: 879569696
Author
Parents
Loading