math

JAX visitors for math expressions.

Sin, Cos, Tan, Square, Sqrt, Exp, Log, Abs, Max,

PositivePart, Huber, SmoothReLU, LogSumExp, Linterp, Bilerp