Public API:
nt.stax
requires()
nt.empirical
nt.predict
nt.batch
nt.monte_carlo_kernel_fn
Internal:
nt.experimental
Kernel
Colab Examples:
Papers:
Other Resources:
Returns a decorator that augments kernel_fn with consistency checks.
kernel_fn
Use this to specify your kernel_fn input kernel requirements.
See also
Diagonal, Bool.
Diagonal
Bool