Alternatives and similar repositories for esmjax:

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

• fattorib / transformer_shmap
  Tensor Parallelism with JAX + Shard Map
  ☆11Updated last year
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders in JAX.
  ☆72Updated 3 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆189Updated last month
• jarridrb / DEM
  Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.
  ☆46Updated this week
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆135Updated 11 months ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆53Updated 10 months ago
• instadeepai / protein-sequence-bfn
  Supporting code for our paper "Protein Sequence Modelling with Bayesian Flow Networks"
  ☆21Updated 3 months ago
• yobibyte / ml-drug-discovery-tldrs
  TLDRs for ML in Drug Discovery papers
  ☆71Updated last year
• wgrathwohl / GWG_release
  Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"
  ☆54Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated last month
• brandontrabucco / design-bench
  Benchmarks for Model-Based Optimization
  ☆84Updated 9 months ago
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆259Updated last week
• e3nn / e3nn-jax
  jax library for E3 Equivariant Neural Networks
  ☆195Updated last week
• emilemathieu / escnn_jax
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆29Updated last year
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆232Updated last year
• google-deepmind / md4
  Code for https://arxiv.org/abs/2406.04329
  ☆56Updated last month
• MJ10 / BioSeq-GFN-AL
  Code for "Biological Sequence Design with GFlowNets", 2022
  ☆72Updated last year
• sail-sg / autofd
  Automatic Functional Differentiation in JAX
  ☆61Updated 2 weeks ago
• samuelstanton / lambo
  Code to reproduce experiments in "Accelerating Bayesian Optimization for Protein Design with Denoising Autoencoders" (Stanton et al 2022)
  ☆67Updated 8 months ago
• graphcore-research / pyscf-ipu
  PySCF on IPU
  ☆42Updated 10 months ago
• lucidrains / invariant-point-attention
  Implementation of Invariant Point Attention, used for coordinate refinement in the structure module of Alphafold2, as a standalone Pytorc…
  ☆157Updated 2 years ago
• ngruver / NOS
  Protein Design with Guided Discrete Diffusion
  ☆123Updated 11 months ago
• jax-ml / jax-ai-stack
  ☆114Updated this week
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆23Updated last month
• alexhernandezgarcia / gflownet
  Generative Flow Networks - GFlowNet
  ☆221Updated 2 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆129Updated 2 months ago
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆17Updated last year
• HannesStark / dirichlet-flow-matching
  ☆96Updated 8 months ago
• IntelLabs / EquiTriton
  EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…
  ☆53Updated last week
• davisyoshida / haiku-mup
  A port of muP to JAX/Haiku
  ☆25Updated 2 years ago