math

JAX visitors for math expressions.

Sin, Cos, Tan, Asin, Acos, Atan, Atan2, Square, Sqrt, Exp, Log, Abs,

Max, Min, PositivePart, Huber, SmoothReLU, LogSumExp, Linterp, Cinterp, Bilerp