ml4physics / cyjax

Related projects ⓘ

Alternatives and complementary repositories for cyjax

• pythoncymetric / cymetric
  A python library for studying Calabi-Yau metrics
  ☆23Updated 3 months ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆80Updated last week
• NIFTy-PPL / JAXbind
  Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
  ☆61Updated 4 months ago
• johannbrehmer / pointers
  Things that make me feel productive
  ☆15Updated 2 years ago
• google-deepmind / flows_for_atomic_solids
  ☆48Updated 2 years ago
• NIFTy-PPL / NIFTy
  Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions
  ☆21Updated this week
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆105Updated last week
• y1xiaoc / fwdlap
  Forward mode laplacian implemented in JAX tracer
  ☆29Updated 2 weeks ago
• fizisist / LorentzGroupNetwork
  Neural network architecture that is fully equivariant with respect to transformations under the Lorentz group, a fundamental symmetry of …
  ☆48Updated 2 years ago
• lukasheinrich / pyhep2020-autodiff-tutorial
  ☆31Updated 4 years ago
• madjax-hep / madjax
  madjax
  ☆14Updated this week
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆135Updated 3 weeks ago
• sbi-benchmark / sbibm
  Simulation-based inference benchmark
  ☆91Updated this week
• DifferentiableUniverseInitiative / jax_cosmo
  A differentiable cosmology library in JAX
  ☆178Updated 3 months ago
• astro-informatics / s2fft
  Differentiable and accelerated spherical transforms with JAX
  ☆136Updated this week
• iaifi / summer-school-2023
  ☆21Updated last year
• marylou-gabrie / flonaco
  ☆19Updated 2 years ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆66Updated last year
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆44Updated 3 weeks ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• gradhep / relaxed
  Provides differentiable versions of common HEP operations and objectives.
  ☆24Updated last year
• Joshuaalbert / jaxns
  Probabilistic Programming and Nested sampling in JAX
  ☆148Updated last week
• dfm / jpu
  JAX + Units
  ☆43Updated last week
• dirmeier / sbijax
  Simulation-based inference in JAX
  ☆21Updated last month
• florpi / summer_school_generative
  ☆13Updated last year
• microsoft / folx
  Implementation of Forward Laplacian algorithm in JAX
  ☆55Updated 2 weeks ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆82Updated 11 months ago
• alexOarga / haiku-geometric
  A collection of graph neural networks implementations in JAX
  ☆31Updated 11 months ago
• niklasschmitz / ad-kernels
  Code for the paper "Algorithmic Differentiation for Automatized Modelling of Machine Learned Force Fields"
  ☆12Updated last year
• thoglu / jammy_flows
  A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…
  ☆42Updated 9 months ago