Jake VanderPlas e9acaa8484 Remove the initial argument to jax.nn.softmax and jax.nn.log_softmax. ...

This argument was deprecated in JAX v0.4.27 and has no effect in JAX v0.4.27 and later.

PiperOrigin-RevId: 693023366

2024-11-04 10:55:21 -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

Remove the initial argument to jax.nn.softmax and jax.nn.log_softmax.

2024-11-04 10:55:21 -08:00

initializers.py

update README and several docs to typed RNG keys

2024-08-11 08:09:47 -07:00