"""
Various functions that may be useful in component compute methods.

These functions are implemented using jax.numpy for the underlying calculations.
This will allow these functions to be used in a way that supports jax's automatic differentiation
and just-in-time compilation capabilities.
"""
from .smooth import act_tanh, smooth_max, smooth_min, smooth_abs, smooth_round
from .ks import ks_max, ks_min
