To use Dropout properly in Equinox, pass a key during training and generate a new one each step with jax.random.split. No key is needed during inference.
key
jax.random.split