ziatdinovmax / NeuroBayes
Fully and Partially Bayesian Neural Nets
☆68Updated 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☆222Updated 6 months ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆233Updated last week
- Probabilistic Programming and Nested sampling in JAX☆179Updated 3 weeks ago
- A system for scientific simulation-based inference at scale.☆164Updated last year
- ☆158Updated 2 months ago
- Numerical quadrature with JAX☆58Updated this week
- Gaussian Mixture Models in Jax☆18Updated 3 months ago
- Simulation-based inference in JAX☆31Updated 3 weeks ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆103Updated last year
- Likelihood-free AMortized Posterior Estimation with PyTorch☆127Updated 8 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆67Updated 2 weeks ago
- Multiple dispatch over abstract array types in JAX.☆120Updated last week
- Matrix-free linear algebra in JAX.☆118Updated 4 months ago
- Notebook to go along with a lecture for the MIT course 8.16: Data Science in Physics on neural simulation-based inference.☆55Updated last year
- S2FFT: Differentiable and accelerated spherical transforms☆150Updated this week
- Minimal Implementation of Bayesian Optimization in JAX☆95Updated 2 weeks ago
- Probabilistic modeling of tabular data with normalizing flows.☆56Updated 2 weeks ago
- A software package for flexible HPC GPs☆16Updated last week
- Interpolation and function approximation with JAX☆177Updated 2 weeks ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated last month
- A differentiable cosmology library in JAX☆196Updated 4 months ago
- Orthogonal polynomials with JAX☆22Updated this week
- Stencil computations in JAX☆70Updated last year
- Unleash the true power of scheduling☆29Updated last month
- ☆19Updated 3 years ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆32Updated 2 months ago
- JAX + Units☆47Updated last week
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆91Updated this week
- Community-sourced list of papers and resources on neural simulation-based inference.☆116Updated last week