softclip (pypi): JAX/Flax implementation of softclip
Keywords: jax