ziatdinovmax / NeuroBayes
Fully and Partially Bayesian Neural Nets
☆64Updated last 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☆221Updated 4 months ago
- Minimal Implementation of Bayesian Optimization in JAX☆89Updated 2 months ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆223Updated this week
- Probabilistic Programming and Nested sampling in JAX☆171Updated 3 months ago
- Matrix-free linear algebra in JAX.☆116Updated 2 months ago
- ☆150Updated 2 weeks ago
- Stencil computations in JAX☆70Updated last year
- Simulation-based inference in JAX☆31Updated last month
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated last month
- Numerical quadrature with JAX☆53Updated last week
- Likelihood-free AMortized Posterior Estimation with PyTorch☆125Updated 6 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆66Updated 8 months ago
- A system for scientific simulation-based inference at scale.☆163Updated 11 months ago
- Interpolation and function approximation with JAX☆163Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆89Updated 2 weeks ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆101Updated last year
- Differentiable and accelerated spherical transforms with JAX☆146Updated this week
- JAX + Units☆47Updated last month
- Multiple dispatch over abstract array types in JAX.☆115Updated 3 weeks ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆124Updated 6 months ago
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆43Updated 2 weeks ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆103Updated 4 months ago
- Community-sourced list of papers and resources on neural simulation-based inference.☆110Updated last month
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆60Updated last month
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- A software package for flexible HPC GPs☆15Updated 2 weeks ago
- Normalizing Flows using JAX☆82Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆105Updated last week
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆30Updated 2 weeks ago
- Notebook to go along with a lecture for the MIT course 8.16: Data Science in Physics on neural simulation-based inference.☆52Updated last year