Alternatives and similar repositories for s2wav:

Users that are interested in s2wav are comparing it to the libraries listed below

• francois-rozet / inox
  Stainless neural networks in JAX
  ☆34Updated last month
• martin-marek / mini-hmc-jax
  A simple implementation of Hamiltonian Monte Carlo in JAX.
  ☆19Updated last year
• smsharma / jax-conditional-flows
  Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.
  ☆20Updated last year
• EddieCunningham / generax
  Generative models with JAX.
  ☆17Updated 4 months ago
• AlexandreAdam / score_models
  A repository to store state of the art score model architectures
  ☆49Updated 5 months ago
• francois-rozet / diffusion-priors
  Official implementation of Learning Diffusion Priors from Observations by Expectation Maximization
  ☆24Updated last month
• homerjed / sbiax
  Fast, lightweight and parallelised simulation-based inference in JAX.
  ☆18Updated this week
• dingo-gw / flow-matching-posterior-estimation
  ☆25Updated last year
• Ciela-Institute / PQM
  Implemenation of PQMass from Lemos et al. 2024
  ☆15Updated 2 weeks ago
• hsimonfroy / benchmark-field-level
  Benchmarking field-level cosmological inference from galaxy surveys.
  ☆10Updated last week
• dirmeier / sbijax
  Simulation-based inference in JAX
  ☆31Updated 3 weeks ago
• GalacticDynamics / unxt
  Unitful Quantities in JAX
  ☆37Updated this week
• DifferentiableUniverseInitiative / JaxPM
  JAX-powered Cosmological Particle-Mesh N-body Solver
  ☆40Updated 2 months ago
• smsharma / lensing-neural-fields
  Implicit neural representations for strong lensing source reconstruction. Code repository associated with https://arxiv.org/abs/2206.1482…
  ☆11Updated 2 years ago
• jfcrenshaw / pzflow
  Probabilistic modeling of tabular data with normalizing flows.
  ☆56Updated 2 weeks ago
• Ciela-Institute / tarp
  Coverage tests to check the quality of your posterior estimators.
  ☆30Updated 8 months ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆91Updated this week
• CosmoStat / jax-lensing
  A JAX package for weak gravitational lensing
  ☆31Updated last year
• tomcharnock / IMNN
  Using neural networks to extract sufficient statistics from data by maximising the Fisher information
  ☆32Updated last year
• NIFTy-PPL / JAXbind
  Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
  ☆67Updated 2 weeks ago
• GalSim-developers / JAX-GalSim
  JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.
  ☆29Updated this week
• Pablo-Lemos / GGNS
  Gradient Based Nested Sampling
  ☆20Updated last year
• DifferentiableUniverseInitiative / jaxDecomp
  JAX bindings for the NVIDIA cuDecomp library
  ☆35Updated this week
• DifferentiableUniverseInitiative / sbi_lens
  A python package for Weak Lensing Implicit Inference with Differentiable Simulator
  ☆15Updated last year
• DifferentiableUniverseInitiative / GalFlow
  A TensorFlow reimplementation of GalSim
  ☆10Updated 3 years ago
• EiffL / Quarks2CosmosDataChallenge
  Material for the CMU Quarks2Cosmos Conference Data Challenges
  ☆14Updated last year
• JAXtronomy / spexial
  Special functions in Jax
  ☆13Updated 2 months ago
• uofgravity / glasflow
  Normalising flows implemented using nflows
  ☆27Updated 7 months ago
• JakobRobnik / MicroCanonicalHMC
  MCHMC: sampler from an arbitrary differentiable distribution
  ☆73Updated last week
• mackelab / simformer
  Combination of transformers and diffusion models for flexible all-in-one simulation-based inference
  ☆65Updated 10 months ago