Alternatives and similar repositories for haiku-geometric

Users that are interested in haiku-geometric are comparing it to the libraries listed below

• beckermr / shape-measurement-tutorials

  View on GitHub
  tutorials on shape measurement
  ☆10Jan 5, 2022Updated 4 years ago
• VMBoehm / MADLens

  View on GitHub
  a differentiable lensing simulator
  ☆10Jan 18, 2022Updated 4 years ago
• EiffL / JaxKDTree

  View on GitHub
  A Jax wrapper for cudaKDTree
  ☆11Sep 26, 2025Updated 4 months ago
• EiffL / LLMAssistedResearch

  View on GitHub
  Exploring how ChatGPT can be used to accelerate research in cosmology.
  ☆12Dec 12, 2022Updated 3 years ago
• dfm / emcee-jax

  View on GitHub
  An experiment: emcee implemented in JAX
  ☆28Jul 1, 2022Updated 3 years ago
• ohahn / DISCO-EB

  View on GitHub
  JAX-based linear Einstein-Boltzmann solver for cosmology
  ☆24Nov 13, 2025Updated 3 months ago
• tomcharnock / IMNN

  View on GitHub
  Using neural networks to extract sufficient statistics from data by maximising the Fisher information
  ☆32May 12, 2023Updated 2 years ago
• dpiras / cosmopower-jax

  View on GitHub
  Differentiable cosmological emulators: the JAX version of CosmoPower
  ☆43Nov 27, 2025Updated 2 months ago
• DifferentiableUniverseInitiative / JaxPM

  View on GitHub
  JAX-powered Cosmological Particle-Mesh N-body Solver
  ☆58Dec 15, 2025Updated 2 months ago
• e3nn / e3nn-jax

  View on GitHub
  jax library for E3 Equivariant Neural Networks
  ☆224Aug 25, 2025Updated 5 months ago
• Herculens / herculens

  View on GitHub
  Differentiable strong lens modelling on CPUs and GPUs
  ☆26Updated this week
• lenstronomy / JAXtronomy

  View on GitHub
  JAX port of lenstronomy, for parallelized, GPU accelerated, and differentiable gravitational lensing and image simulations.
  ☆20Updated this week
• AngusWright / AstroStats

  View on GitHub
  Repository for Angus H Wright's AstroStatistics Lecture Materials
  ☆15Apr 6, 2022Updated 3 years ago
• DifferentiableUniverseInitiative / jaxDecomp

  View on GitHub
  JAX reimplementation and bindings for the NVIDIA cuDecomp library
  ☆52Updated this week
• cbottrell / RealSim

  View on GitHub
  RealSim is the statistical observational realism suite described in Bottrell et al 2017ab and made public in Bottrell et al 2019b.
  ☆12Jul 7, 2021Updated 4 years ago
• hsimonfroy / benchmark-field-level

  View on GitHub
  Benchmarking field-level cosmological inference from galaxy surveys.
  ☆12Jul 17, 2025Updated 6 months ago
• DifferentiableUniverseInitiative / GalFlow

  View on GitHub
  A TensorFlow reimplementation of GalSim
  ☆10Mar 8, 2022Updated 3 years ago
• LSSTDESC / descwl-shear-sims

  View on GitHub
  simple simulations for testing weak lensing shear measurement
  ☆12Oct 31, 2025Updated 3 months ago
• tlmakinen / cosmicGraphs

  View on GitHub
  Investigating the information content in the cosmic web through dark matter halo graphs. Arxiv paper: https://arxiv.org/abs/2207.05202
  ☆13Jan 18, 2024Updated 2 years ago
• smsharma / lensing-neural-fields

  View on GitHub
  Implicit neural representations for strong lensing source reconstruction. Code repository associated with https://arxiv.org/abs/2206.1482…
  ☆12Feb 26, 2023Updated 2 years ago
• gerkone / egnn-jax

  View on GitHub
  E(n) Equivariant GNN in jax
  ☆14Aug 31, 2023Updated 2 years ago
• yuyang-shi / generalized-diffusion

  View on GitHub
  ☆16May 30, 2023Updated 2 years ago
• LSSTScienceCollaborations / StackClubCourse

  View on GitHub
  Repository for course offered by the Stack Club
  ☆13Jan 13, 2023Updated 3 years ago
• langosco / neural-variational-gradient-descent

  View on GitHub
  Code for the paper 'Neural Variational Gradient Descent'. We perform nonparametric variational inference by transporting samples along a …
  ☆12Jul 29, 2021Updated 4 years ago
• GalSim-developers / JAX-GalSim

  View on GitHub
  JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.
  ☆33Updated this week
• zhao-david / CDE-diagnostics

  View on GitHub
  Diagnostics for Conditional Density Estimators and Bayesian Inference Algorithms
  ☆15Apr 3, 2022Updated 3 years ago
• mancusolab / traceax

  View on GitHub
  Stochastic trace estimation using JAX
  ☆17Aug 20, 2025Updated 5 months ago
• modichirag / GSM-VI

  View on GitHub
  Code for Gaussian Score Matching Variational Inference
  ☆35Feb 23, 2025Updated 11 months ago
• gcucurull / jax-gcn

  View on GitHub
  Graph Convolutional Networks in JAX
  ☆33Jan 3, 2021Updated 5 years ago
• DifferentiableUniverseInitiative / jax-cosmo-paper

  View on GitHub
  Paper introducing jax-cosmo
  ☆13Apr 27, 2023Updated 2 years ago
• esheldon / metadetect

  View on GitHub
  Library for meta-detection, combining detection and metacalibration
  ☆13Jan 23, 2026Updated 3 weeks ago
• montefiore-institute / balanced-nre

  View on GitHub
  Code for the paper "Towards Reliable Simulation-Based Inference with Balanced Neural Ratio Estimation".
  ☆14Nov 14, 2022Updated 3 years ago
• itsdaniele / jeometric

  View on GitHub
  Graph neural networks in JAX.
  ☆68Jun 18, 2024Updated last year
• CosmoStat / shapepipe

  View on GitHub
  Shape Measurement Pipeline
  ☆17Updated this week
• LSSTDESC / rail_attic

  View on GitHub
  Redshift Assessment Infrastructure Layers
  ☆14Aug 1, 2023Updated 2 years ago
• fidero / gp-derivative

  View on GitHub
  Example code to speed up GP inference with gradients for high-dimensional inputs.
  ☆14Jul 12, 2021Updated 4 years ago
• marius311 / Cosmology.jl

  View on GitHub
  Compute cosmological quantities in Julia
  ☆13Dec 14, 2020Updated 5 years ago
• CosmoStat / jax-lensing

  View on GitHub
  A JAX package for weak gravitational lensing
  ☆35Apr 30, 2024Updated last year
• leonard-gleyzer / connex

  View on GitHub
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Jul 23, 2023Updated 2 years ago