Code to accompany paper 'Bayesian Deep Ensembles via the Neural Tangent Kernel'
☆26Mar 12, 2021Updated 4 years ago
Alternatives and similar repositories for bayesian-ntk
Users that are interested in bayesian-ntk are comparing it to the libraries listed below
Sorting:
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- Code accompanying our paper "Finding trainable sparse networks through Neural Tangent Transfer" to be published at ICML-2020.☆13Jun 14, 2020Updated 5 years ago
- Learning with Higher Expressive Power than Neural Networks (On Learning PDEs)☆16Feb 17, 2021Updated 5 years ago
- The source code the for the ICLR'24 paper "Stabilizing Backpropagation Through Time to Learn Complex Physics"☆11May 17, 2024Updated last year
- Finding the conditional distributions of a Gaussian Mixture Model☆13Nov 10, 2019Updated 6 years ago
- Public Repo for the paper "Overcoming The Spectral-Bias of Neural Value Approximation"☆11May 25, 2024Updated last year
- Repository for tutorial on Neural ODEs prepared for the UCL AI Society☆13Mar 7, 2021Updated 4 years ago
- Implementation of Physics-Informed PointNet (PIPN) for weakly-supervised learning of 2D linear elasticity (plane stress) on multiple sets…☆12Apr 11, 2024Updated last year
- Conditional Flow Matching using JAX☆13Jun 22, 2023Updated 2 years ago
- Generative turbulence model TurbDiff as proposed in "From Zero to Turbulence: Generative Modeling for 3D Flow Simulation", ICLR 2024☆34Dec 7, 2025Updated 2 months ago
- ☆68Mar 15, 2019Updated 6 years ago
- ☆19Apr 5, 2024Updated last year
- Inducing Point Operator Transformer: A Flexible and Scalable Architecture for Solving PDEs (AAAI 2024)☆15Jul 30, 2024Updated last year
- NeurIPS paper 'Censored Quantile Regression Neural Networks for Distribution-Free Survival Analysis'☆11Oct 28, 2022Updated 3 years ago
- Official Implementation of "AIVT: Inference of turbulent thermal convection from measured 3D velocity data by physics-informed Kolmogorov…☆16Oct 10, 2025Updated 4 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆57May 23, 2023Updated 2 years ago
- Official PyTorch implementation of "Energy-Based Contrastive Learning of Visual Representations", NeurIPS 2022 Oral Paper☆13Oct 2, 2022Updated 3 years ago
- ☆13Nov 30, 2023Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- Implementation of deep kernels in GPyTorch and Pyro☆12Jan 19, 2021Updated 5 years ago
- Code repo for "Function-Space Distributions over Kernels"☆32Jan 21, 2021Updated 5 years ago
- Code for random Fourier features based on Rahimi and Recht's 2007 paper.☆58Dec 21, 2020Updated 5 years ago
- Implementation of Rational Kolmogorov-Arnold Network (rKAN)☆19Jun 28, 2024Updated last year
- Code for "Robust flow field reconstruction from limited measurements vis sparse representation" (J. Callaham, K. Maeda, and S. Brunton 20…☆14Oct 23, 2018Updated 7 years ago
- Improving predictions of Bayesian neural nets via local linearization, AISTATS 2021☆16Dec 30, 2022Updated 3 years ago
- ☆12Aug 22, 2025Updated 6 months ago
- Unofficial implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, using Flax with the Linen API☆13Sep 25, 2021Updated 4 years ago
- Synthetic data library used in operator learning for PDE problems that overcomes dependence on classical solvers such as finite differenc…☆17Aug 8, 2024Updated last year
- Code used for experiments in https://arxiv.org/abs/2008.08601☆19Jun 1, 2021Updated 4 years ago
- ☆26Jun 12, 2025Updated 8 months ago
- ☆19Nov 24, 2024Updated last year
- ☆15Mar 3, 2023Updated 3 years ago
- About Code Release for "Solving High-Dimensional PDEs with Latent Spectral Models" (ICML 2023), https://arxiv.org/abs/2301.12664☆79Apr 10, 2025Updated 10 months ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 4 years ago
- Official code for the ICLR 2021 paper Neural ODE Processes☆76May 1, 2022Updated 3 years ago
- An elegant adaptive importance sampling algorithms for simulations of multi-modal distributions (NeurIPS'20)☆42May 20, 2022Updated 3 years ago
- Implementation of the paper: "FedTabDiff: Federated Learning of Diffusion Models for Synthetic Mixed-Type Tabular Data Generation"☆21Nov 10, 2024Updated last year
- ☆22Oct 23, 2020Updated 5 years ago
- ☆20Mar 14, 2021Updated 4 years ago