DeaglanBartlett / ESR
☆24Updated 5 months ago
Alternatives and similar repositories for ESR:
Users that are interested in ESR are comparing it to the libraries listed below
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆64Updated 7 months ago
- Python bindings and scikit-learn interface for the Operon library for symbolic regression.☆44Updated last month
- Numerical quadrature with JAX☆50Updated this week
- JAX bindings for the NVIDIA cuDecomp library☆32Updated this week
- Orthogonal polynomials with JAX☆12Updated this week
- ☆31Updated 4 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆84Updated 3 weeks ago
- Efficient Differentiable n-d PDE solvers in JAX.☆23Updated 2 months ago
- Using Graph Neural Networks to regress baryonic properties directly from full dark matter merger trees.☆24Updated last year
- ☆21Updated last year
- [DMLR] Rethinking Symbolic Regression Datasets and Benchmarks for Scientific Discovery☆32Updated 10 months ago
- ☆15Updated 9 months ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆34Updated last week
- Turning SymPy expressions into JAX functions☆43Updated 3 years ago
- ☆13Updated last year
- Special functions in Jax☆12Updated 3 weeks ago
- A lightweight, parallel python wrapper for the GSL integration library☆14Updated 3 years ago
- Graph Neural Networks to predict the cosmological parameters or the galaxy power spectrum from galaxy catalogs.☆15Updated 2 years ago
- Finite difference tools in JAX☆17Updated 11 months ago
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆57Updated last week
- Next generation Bayesian analysis tools for efficient posterior sampling and evidence estimation.☆29Updated 2 years ago
- Stencil computations in JAX☆70Updated last year
- A software package for flexible HPC GPs☆14Updated 2 months ago
- A collection of graph neural networks implementations in JAX☆30Updated last year
- Simulation-based inference in JAX☆29Updated last week
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆22Updated this week
- Gradient Based Nested Sampling☆19Updated 11 months ago
- A Jax wrapper for cudaKDTree☆10Updated last year
- JAX + Units☆44Updated this week
- MCHMC: sampler from an arbitrary differentiable distribution☆72Updated this week