Skip to content

Remove casting from jax.nn.one_hot #705

Remove casting from jax.nn.one_hot

Remove casting from jax.nn.one_hot #705

TPU test (jaxlib=head, v5e-8)

succeeded Dec 23, 2024 in 5m 59s