ziatdinovmax / NeuroBayes
Fully and Partially Bayesian Neural Nets
☆52Updated this week
Alternatives and similar repositories for NeuroBayes:
Users that are interested in NeuroBayes are comparing it to the libraries listed below
- Gaussian Processes for Experimental Sciences☆218Updated 3 months ago
- Probabilistic Programming and Nested sampling in JAX☆166Updated 2 months ago
- A system for scientific simulation-based inference at scale.☆163Updated 10 months ago
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆87Updated last week
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆209Updated this week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆65Updated 7 months ago
- Implementation of Forward Laplacian algorithm in JAX☆59Updated 2 weeks ago
- Minimal Implementation of Bayesian Optimization in JAX☆88Updated last month
- Notebook to go along with a lecture for the MIT course 8.16: Data Science in Physics on neural simulation-based inference.☆50Updated last year
- Interpolation and function approximation with JAX☆157Updated this week
- ☆141Updated this week
- Differentiable and accelerated spherical transforms with JAX☆145Updated this week
- A differentiable cosmology library in JAX☆194Updated last month
- Stencil computations in JAX☆70Updated last year
- Numerical quadrature with JAX☆53Updated last week
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆102Updated 3 months ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆122Updated 5 months ago
- Matrix-free linear algebra in JAX.☆114Updated last month
- Simulation-based inference in JAX☆29Updated last week
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆42Updated last year
- ☆48Updated 2 years ago
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆59Updated last week
- Multiple dispatch over abstract array types in JAX.☆114Updated last week
- Unleash the true power of scheduling☆27Updated last week
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆104Updated 2 weeks ago
- Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines☆10Updated last year
- JAX + Units☆45Updated 2 weeks ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆30Updated last week
- Community-sourced list of papers and resources on neural simulation-based inference.☆108Updated 3 weeks ago