Public API:
nt.stax
Monomial()
nt.empirical
nt.predict
nt.batch
nt.monte_carlo_kernel_fn
Internal:
nt.experimental
Kernel
Colab Examples:
Papers:
Other Resources:
Monomials, i.e. x^degree.
x^degree
degree (int) – an integer between 0 and 5.
int
tuple[InitFn, ApplyFn, LayerKernelFn]
tuple
InitFn
ApplyFn
LayerKernelFn
(init_fn, apply_fn, kernel_fn).