Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processes"
☆252Aug 26, 2020Updated 5 years ago
Alternatives and similar repositories for GP4A
Users that are interested in GP4A are comparing it to the libraries listed below
Sorting:
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆110Aug 26, 2020Updated 5 years ago
- Convolutional Neural Tangent Kernel☆112Oct 22, 2019Updated 6 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,377Mar 1, 2024Updated 2 years ago
- Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube☆48Sep 19, 2019Updated 6 years ago
- NeurIPS 2019 Paper Implementation☆12Nov 22, 2022Updated 3 years ago
- Reference implementation of algorithms for reinforcement learning and Markov decision processes.☆12Jan 28, 2021Updated 5 years ago
- Deep neural network kernel for Gaussian process☆211Sep 8, 2020Updated 5 years ago
- ☆45Nov 3, 2019Updated 6 years ago
- Software relating to relational empirical risk minimization☆17Jun 12, 2021Updated 4 years ago
- ☆35Jul 19, 2021Updated 4 years ago
- Official Code for Invertible Residual Networks☆536Jul 28, 2024Updated last year
- ☆12Dec 20, 2019Updated 6 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆129Jul 14, 2024Updated last year
- Benchmark for LP-relaxed robustness verification of ReLU-networks☆42Apr 24, 2019Updated 6 years ago
- Pytorch implementation of Block Neural Autoregressive Flow☆182Aug 19, 2021Updated 4 years ago
- PyTorch implementations of algorithms for density estimation☆586May 13, 2021Updated 4 years ago
- Normalizing flows in PyTorch. Current intended use is education not production.☆908Jan 27, 2020Updated 6 years ago
- Understanding Training Dynamics of Deep ReLU Networks☆306Oct 19, 2025Updated 4 months ago
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆25Nov 2, 2021Updated 4 years ago
- A general, modular, and programmable architecture search framework☆124Mar 24, 2023Updated 2 years ago
- Implementation of Graph Neural Tangent Kernel (NeurIPS 2019)☆105Jan 28, 2020Updated 6 years ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,627Mar 25, 2022Updated 3 years ago
- This repository contains notebook implementations of the following Neural Process variants: Conditional Neural Processes (CNPs), Neural P…☆1,015Jan 19, 2021Updated 5 years ago
- Notebooks for reproducing the paper "Computer Vision with a Single (Robust) Classifier"☆129Oct 24, 2019Updated 6 years ago
- List of awesome JAX resources☆13Dec 8, 2022Updated 3 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Dec 27, 2022Updated 3 years ago
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,123Apr 20, 2022Updated 3 years ago
- Randomized Smoothing of All Shapes and Sizes (ICML 2020).☆51Jul 23, 2020Updated 5 years ago
- Code for experiments in my blog post on the Neural Tangent Kernel: https://eigentales.com/NTK☆173Nov 10, 2019Updated 6 years ago
- Differentiable convex optimization layers☆2,063Feb 27, 2026Updated last week
- Source code for ICLR 2020 paper: "Learning to Guide Random Search"☆39Sep 2, 2024Updated last year
- ☆153May 25, 2020Updated 5 years ago
- Pytorch implementations of density estimation algorithms: BNAF, Glow, MAF, RealNVP, planar flows☆637Jul 12, 2021Updated 4 years ago
- [NeurIPS 2019 Spotlight] High dimensional mean estimation and outlier detection in nearly-linear time.☆26Oct 11, 2019Updated 6 years ago
- Shape and dimension inference (Keras-like) for PyTorch layers and neural networks☆570Jun 13, 2022Updated 3 years ago
- Large-scale, multi-GPU capable, kernel solver☆197Jul 17, 2025Updated 7 months ago
- Code for Implicit Generation and Generalization with Energy Based Models☆363Apr 25, 2023Updated 2 years ago
- Source code for "On the Relationship between Self-Attention and Convolutional Layers"☆1,117Jan 10, 2023Updated 3 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,159Feb 6, 2026Updated last month