trax-extra-activation (pypi): Extra activation functions for Trax
Keywords: trax|jax|machine learning