tmdsimpy.jax.harmonic_utils

Subset of tmdsimpy.utils.harmonic using JAX

Autodiff is applied mainly for AFT at this point, thus only some functions need to be converted to JAX. Other functions are not updated at this point to use JAX.

See Also

tmdsimpy.utils.harmonic :

Baseline implementation of harmonic utility methods without JAX.

Functions

get_fourier_coeff(htuple, x_t)	Calculates a specific set of Fourier coefficients of a time series.
time_series_deriv(Nt, htuple, X0, order)	Returns derivative of a time series defined by a set of harmonics