https://github.com/google/jax
Raw File
Tip revision: 3b0db5c90027bcf2c883bc56a9a1d0382579e8f9 authored by Jake VanderPlas on 16 May 2024, 18:04:24 UTC
jax.nn.softmax: only use initial=-inf when necessary
Tip revision: 3b0db5c
JAX_demo.ipynb
back to top