DeaglanBartlett / ESR
☆22Updated last month
Related projects: ⓘ
- Python bindings and scikit-learn interface for the Operon library for symbolic regression.☆38Updated 2 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 2 months ago
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆54Updated last month
- Numerical quadrature with JAX☆39Updated last month
- Using Graph Neural Networks to regress baryonic properties directly from full dark matter merger trees.☆24Updated last year
- Generative models with JAX.☆18Updated 8 months ago
- A paper describing the implementation of PySR and SymbolicRegression.jl☆45Updated 7 months ago
- MCHMC: sampler from an arbitrary differentiable distribution☆69Updated this week
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- [DMLR] Rethinking Symbolic Regression Datasets and Benchmarks for Scientific Discovery☆26Updated 6 months ago
- Interpolation and function approximation with JAX☆117Updated 2 weeks ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆77Updated last week
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆188Updated last week
- Probabilistic Programming and Nested sampling in JAX☆141Updated 3 weeks ago
- ☆100Updated 3 years ago
- Stainless neural networks in JAX☆30Updated 2 weeks ago
- Probabilistic modeling of tabular data with normalizing flows.☆54Updated 3 months ago
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆16Updated this week
- ☆85Updated this week
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆21Updated 4 months ago
- ☆31Updated 3 years ago
- A differentiable cosmology library in JAX☆172Updated last month
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt☆45Updated this week
- A Python package for approximate Bayesian inference and optimization using Gaussian processes☆41Updated 10 months ago
- Minimal Implementation of Bayesian Optimization in JAX☆82Updated 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.☆42Updated 10 months ago
- ☆18Updated 10 months ago
- JAX bindings for the NVIDIA cuDecomp library☆24Updated 2 weeks ago
- A spherical convolutional neural network for cosmology (TFv2).☆19Updated 5 months ago
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆20Updated 8 months ago