Jake VanderPlas a282d586b6 nn.softmax: use double-where when where is specified

2024-01-26 09:45:31 -08:00

..

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

functions.py

nn.softmax: use double-where when where is specified

2024-01-26 09:45:31 -08:00

initializers.py

Decorate jax.nn.initializers.Initializer as @typing.runtime_checkable

2024-01-15 05:44:18 -08:00