Alternatives and similar repositories for Jax-Wavelet-Toolbox

Users that are interested in Jax-Wavelet-Toolbox are comparing it to the libraries listed below

• astro-informatics / QuantifAI

  View on GitHub
  PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…
  ☆12Feb 17, 2025Updated 11 months ago
• carnotresearch / cr-sparse

  View on GitHub
  Functional models and algorithms for sparse signal processing
  ☆97Oct 17, 2023Updated 2 years ago
• AngusWright / AstroStats

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

  View on GitHub
  A Python library that integrates the Curvelet transform into differentiable programming pipelines, such as PyTorch and JAX.
  ☆12Jun 22, 2023Updated 2 years ago
• sniemi / EuclidVisibleInstrument

  View on GitHub
  Euclid Visible Instrument Python package. Includes a simulator and various analysis codes.
  ☆11Mar 25, 2015Updated 10 years ago
• clementpoiret / Equimo

  View on GitHub
  Implementation of Computer Vision Models in JAX (equinox)
  ☆18Jan 15, 2026Updated last month
• duchao0726 / Conditional-SWF

  View on GitHub
  ☆13Jul 25, 2023Updated 2 years ago
• google-deepmind / xarray_jax

  View on GitHub
  ☆31Nov 14, 2025Updated 3 months ago
• florpi / summer_school_generative

  View on GitHub
  ☆14Aug 8, 2023Updated 2 years ago
• pierreglaser / jax-tutorial

  View on GitHub
  ☆16Nov 3, 2022Updated 3 years ago
• francois-rozet / dawgz

  View on GitHub
  Unleash the true power of scheduling
  ☆33Apr 3, 2025Updated 10 months ago
• francois-rozet / inox

  View on GitHub
  Stainless neural networks in JAX
  ☆34Feb 3, 2026Updated last week
• genssi-developer / GenSSI

  View on GitHub
  GenSSI 2.0 - Generating Series for testing Structural Identifiability
  ☆15Apr 6, 2024Updated last year
• pkmtum / Semi-supervised_Invertible_Neural_Operators

  View on GitHub
  Semi-supervised Invertible Neural Operators for Bayesian Inverse Problems
  ☆15May 27, 2024Updated last year
• 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
• flatironinstitute / pytorch-finufft

  View on GitHub
  Pytorch wrappers for the FINUFFT library
  ☆16Nov 21, 2025Updated 2 months ago
• ohahn / DISCO-EB

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

  View on GitHub
  Python module for interacting with example LISA catalogs
  ☆18Jul 24, 2025Updated 6 months ago
• NIFTy-PPL / JAXbind

  View on GitHub
  Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
  ☆75Updated this week
• francois-rozet / postr

  View on GitHub
  A minimal poster template in Typst.
  ☆26Oct 24, 2025Updated 3 months ago
• homerjed / sbiax

  View on GitHub
  Fast, lightweight and parallelised simulation-based inference in JAX.
  ☆23Oct 28, 2025Updated 3 months ago
• COR-OPT / SuperPolyak.py

  View on GitHub
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Nov 18, 2022Updated 3 years ago
• Joshuaalbert / jaxns

  View on GitHub
  Probabilistic Programming and Nested sampling in JAX
  ☆221Dec 15, 2025Updated 2 months ago
• lanl / scico

  View on GitHub
  Scientific Computational Imaging COde
  ☆150Feb 5, 2026Updated last week
• cjw531 / nerf_tf2

  View on GitHub
  Tensorflow 2 release by @cjw531 for NeRF (Neural Radiance Fields) originally by @bmild
  ☆27Jun 16, 2022Updated 3 years ago
• lisa-analysis-center / glass

  View on GitHub
  Prototype library for LISA data analysis
  ☆31Feb 7, 2026Updated last week
• chenlin9 / Gaussianization_Flows

  View on GitHub
  ☆22May 6, 2020Updated 5 years ago
• cthoyt / autoreviewer

  View on GitHub
  👁️‍🗨️ Scientists often do the same bad stuff. Automate giving deterministic feedback during peer review with determinstic (LLM-free)
  ☆30May 9, 2025Updated 9 months ago
• mikekatz04 / LISAanalysistools

  View on GitHub
  ☆31Updated this week
• tk-rusch / unicornn

  View on GitHub
  Official code for UnICORNN (ICML 2021)
  ☆28Oct 1, 2021Updated 4 years ago
• satwik77 / Transformer-Formal-Languages

  View on GitHub
  EMNLP 2020: On the Ability and Limitations of Transformers to Recognize Formal Languages
  ☆24Oct 10, 2020Updated 5 years ago
• erfanzar / jax-flash-attn2

  View on GitHub
  A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/…
  ☆34Mar 4, 2025Updated 11 months ago
• GalSim-developers / JAX-GalSim

  View on GitHub
  JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.
  ☆32Feb 8, 2026Updated last week
• GeorgeMichailidis / multi-task-mixed-freq

  View on GitHub
  Code repository for "Multi-Task Encoder-Dual-Decoder Modeling Framework on Mixed Frequency Data", International Journal of Forecasting, 2…
  ☆12Feb 18, 2024Updated last year
• mahindrautela / BOPINN

  View on GitHub
  Bayesian optimized physics-informed neural network for parameter estimation
  ☆33Nov 20, 2024Updated last year
• knoxgraeme / skillfish

  View on GitHub
  The skill manager for AI coding agents. Install, update, and sync skills across Claude Code, Cursor, Copilot + more.
  ☆45Updated this week
• CPShub / TutorialMLinSolidMechanics

  View on GitHub
  ☆11Oct 15, 2025Updated 4 months ago
• NIFTy-PPL / NIFTy

  View on GitHub
  Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions
  ☆42Jan 28, 2026Updated 2 weeks ago
• astro-informatics / s2fft

  View on GitHub
  S2FFT: Differentiable and accelerated spherical transforms
  ☆215Updated this week