Neural Tangents
stable
Topics:
Stax – infinite networks (NNGP, NTK)
Combinators
Branching
Linear parametric
Linear nonparametric
Elementwise nonlinear
Helper enums
For developers
neural_tangents.stax.layer
neural_tangents.stax.supports_masking
neural_tangents.stax.requires
neural_tangents.stax.Bool
neural_tangents.stax.Diagonal
Empirical – finite NNGP and NTK
Predict – inference with NNGP and NTK or linearized networks
Batching – using multiple devices
Monte Carlo Sampling
Neural Tangents
»
Stax – infinite networks (NNGP, NTK)
»
neural_tangents.stax.Bool
Edit on GitHub
neural_tangents.stax.Bool
class
neural_tangents.stax.
Bool
(
value
)
[source]
Helper trinary logic class.
__init__
(
)
Attributes
NO
MAYBE
YES
Read the Docs
v: stable
Versions
latest
stable
Downloads
html
On Read the Docs
Project Home
Builds