pytc.df.thc.df_sandwiches_jax¶

pytc.df.thc.df_sandwiches_jax(b, fit, t2, *, rank_panel, aux_panel)[source]¶

Evaluate the panelled production DF/THC contractions.

fit exposes the fitted p_virtual and y factors.

b is deliberately kept HOST-resident (see _as_fp64_host()): the panel loops below only read aux-axis slices, so uploading the whole block would cost its full size in device memory for no benefit.

Parameters:

rank_panel (int)
aux_panel (int)