NIFTy-PPL / NIFTy
Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions
☆21Updated this week
Related projects ⓘ
Alternatives and complementary repositories for NIFTy
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 4 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆80Updated last week
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆22Updated 7 months ago
- Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.☆12Updated last year
- Numerical quadrature with JAX☆44Updated 3 weeks ago
- JAX + Units☆43Updated last week
- ☆19Updated last year
- A Jax wrapper for cudaKDTree☆10Updated last year
- ☆31Updated 4 years ago
- Stencil computations in JAX☆66Updated last year
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆17Updated last year
- Orthogonal polynomials with JAX☆10Updated 3 weeks ago
- JAX bindings for the NVIDIA cuDecomp library☆24Updated 3 weeks ago
- Simulation-based inference in JAX☆21Updated last month
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆41Updated last month
- MCHMC: sampler from an arbitrary differentiable distribution☆70Updated this week
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆30Updated last year
- madjax☆14Updated this week
- Things that make me feel productive☆15Updated 2 years ago
- Differentiable and accelerated spherical transforms with JAX☆136Updated this week
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆12Updated 8 months ago
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆40Updated 4 months ago
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt☆57Updated this week
- Provides differentiable versions of common HEP operations and objectives.☆24Updated last year
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- Gradient-informed particle MCMC methods☆11Updated 9 months ago
- Exponential families for JAX☆55Updated this week
- Evaluate and transform D matrices, 3-j symbols, and (scalar or spin-weighted) spherical harmonics☆45Updated 11 months ago
- Implementation of Arxiv:2212.08549 in Julia☆12Updated 2 months ago
- Notebook to go along with a lecture for the MIT course 8.16: Data Science in Physics on neural simulation-based inference.☆45Updated last year