Public API:
nt.stax
nt.empirical
nt.predict
nt.batch
nt.monte_carlo_kernel_fn
Internal:
nt.experimental
Kernel
Colab Examples:
Papers:
Other Resources:
neural_tangents
neural_tangents._src.batching
neural_tangents._src.empirical
neural_tangents._src.monte_carlo
neural_tangents._src.predict
neural_tangents._src.utils.typing
neural_tangents.experimental
neural_tangents.stax