mish activation function.

[Nightbringers]

please add mish activation function.

[chiamp]

Flax imports activation functions from Jax. Adding the mish activation function was proposed, but ultimately Jax wanted to keep jax.nn very minimal. You could try raising the issue again to see if they're open to adding it, or you can implement it yourself as a custom activation function.