Alternatives and similar repositories for esm2quinox

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

• nboyd / joltz
  JAX translation of boltz
  ☆13Updated 3 months ago
• smsharma / eqnn-jax
  Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.
  ☆33Updated 2 months ago
• prescient-design / e3tools
  Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0
  ☆12Updated last month
• normal-computing / thermox
  Exact OU processes with JAX
  ☆43Updated last month
• plainerman / Variational-Doob
  Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling
  ☆49Updated last month
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 4 months ago
• lollcat / se3-augmented-coupling-flows
  SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.
  ☆24Updated last year
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆24Updated last year
• whitead / minimalaf
  Minimal AlphaFold
  ☆53Updated 8 months ago
• anh-tong / nanoGPT-equinox
  nanoGPT using Equinox
  ☆13Updated 2 years ago
• mancusolab / traceax
  Stochastic trace estimation using JAX
  ☆14Updated last month
• maxxxzdn / jax-geometric
  Implementation of various equivariant models in JAX
  ☆12Updated last year
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆57Updated last year
• patrick-kidger / wadler_lindig
  A Wadler--Lindig pretty printer for Python
  ☆38Updated last week
• alexOarga / haiku-geometric
  A collection of graph neural networks implementations in JAX
  ☆32Updated last year
• prescient-design / holo-bench
  Benchmark for Biophysical Sequence Optimization Algorithms
  ☆16Updated 2 weeks ago
• google-deepmind / annealed_flow_transport
  ☆52Updated 2 years ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆121Updated last week
• PeptoneLtd / nerfax
  An efficient method for the conversion from internal to Cartesian coordinates that utilizes the platform-agnostic JAX Python library.
  ☆17Updated 11 months ago
• jarridrb / DEM
  Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.
  ☆53Updated last month
• lollcat / fab-jax
  Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.
  ☆11Updated 11 months ago
• bb515 / diffusionjax
  diffusionjax is a simple and accessible diffusion models package in JAX
  ☆42Updated 3 months ago
• kksniak / metric-flow-matching
  Official implementation of Metric Flow Matching (NeurIPS 2024)
  ☆35Updated 5 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 5 months ago
• mjo22 / cryojax
  Cryo electron microscopy image simulation and analysis built on JAX.
  ☆34Updated last week
• emilemathieu / escnn_jax
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆30Updated last year
• maxxxzdn / erwin
  Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
  ☆51Updated this week
• cgarciae / simple_flow_matching
  ☆23Updated 4 months ago
• necludov / wl-mechanics
  ☆32Updated 11 months ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated 9 months ago