Search Results for "bayesian"

Neural Tangents is a high-level neural network API for specifying complex, hierarchical models at both finite and infinite width, built in Python on top of JAX and XLA. It lets researchers define architectures from familiar building blocks—convolutions, pooling, residual connections, and nonlinearities—and obtain not only the finite network but also the corresponding Gaussian Process (GP) kernel of its infinite-width limit. With a single specification, you can compute NNGP and NTK kernels,...

...They can learn distributions over functions from data and efficiently make predictions at new inputs with calibrated uncertainty — making them useful for few-shot learning, Bayesian regression, and meta-learning. Each notebook includes theoretical explanations, key building blocks, and executable code that runs directly in Google Colab, requiring no local setup. Implementations rely only on standard dependencies such as NumPy, TensorFlow, and Matplotlib, and provide visualizations of model performance.