ziatdinovmax / NeuroBayes
Fully and Partially Bayesian Neural Nets
☆46Updated 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☆217Updated 3 months ago
- Probabilistic Programming and Nested sampling in JAX☆164Updated last month
- Simulation-based inference in JAX☆29Updated last week
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆208Updated this week
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated 7 months ago
- ☆13Updated last year
- Stencil computations in JAX☆70Updated last year
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆84Updated 3 weeks ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆64Updated 7 months ago
- Differentiable and accelerated spherical transforms with JAX☆143Updated this week
- A system for scientific simulation-based inference at scale.☆163Updated 10 months ago
- Numerical quadrature with JAX☆50Updated this week
- JAX + Units☆44Updated this week
- ☆134Updated last month
- Interpolation and function approximation with JAX☆148Updated this week
- ☆21Updated 5 months ago
- ☆21Updated last year
- Gradient Based Nested Sampling☆19Updated 11 months ago
- Matrix-free linear algebra in JAX.☆113Updated 3 weeks ago
- A differentiable cosmology library in JAX☆188Updated last month
- Unleash the true power of scheduling☆27Updated 2 weeks ago
- Minimal Implementation of Bayesian Optimization in JAX☆88Updated 3 weeks ago
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆22Updated last year
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆57Updated last week
- Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.☆25Updated 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
- Multiple dispatch over abstract array types in JAX.☆114Updated this week
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆98Updated 2 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆99Updated last year
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆56Updated 7 months ago