Alternatives and similar repositories for jax_xc

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

• sail-sg / jrystal
  A JAX-based Differentiable Density Functional Theory Framework for Materials
  ☆41Updated last week
• mdsunivie / deeperwin
  DeepErwin is a python 3.8+ package that implements and optimizes JAX 2.x wave function models for numerical solutions to the multi-electr…
  ☆63Updated 7 months ago
• XanaduAI / GradDFT
  GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine l…
  ☆108Updated last year
• tomdbar / naqs-for-quantum-chemistry
  Supporting code for "Autoregressive neural-network wavefunctions for ab initio quantum chemistry".
  ☆43Updated 3 years ago
• bytedance / netobs
  Calculate observables from neural network-based VMC (NN-VMC).
  ☆17Updated 7 months ago
• fishjojo / pyscfad
  PySCF with auto-differentiation
  ☆89Updated last week
• rubber-duck-debug / cuda-mace
  CUDA implementations of MACE models
  ☆20Updated 3 months ago
• dcccc / Plane_Wave_DFT
  plane wave basis set density functional thoery code
  ☆19Updated 6 months ago
• CCQC / Quax
  Arbitrary-order derivatives of popular electronic structure methods, such as Hartree-Fock and coupled cluster theory.
  ☆70Updated last year
• sail-sg / d4ft
  A JAX library for Density Functional Theory.
  ☆54Updated 2 weeks ago
• SparkyTruck / deepmd-jax
  A lightweight DeepPotentialMD with JAX backend, and more than that! Built for both performance and flexibility in pure Python.
  ☆33Updated 2 weeks ago
• google-research / e3x
  E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.
  ☆109Updated 8 months ago
• diffqc / dqc
  Differentiable Quantum Chemistry (only Differentiable Density Functional Theory and Hartree Fock at the moment)
  ☆120Updated 3 years ago
• graphcore-research / pyscf-ipu
  PySCF on IPU
  ☆44Updated last year
• ACEsuit / mace-jax
  Equivariant machine learning interatomic potentials in JAX.
  ☆80Updated 2 weeks ago
• graphcore-research / mess
  MESS: Modern Electronic Structure Simulations
  ☆20Updated last year
• PASSIONLab / OpenEquivariance
  OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.
  ☆99Updated 3 weeks ago
• microsoft / oneqmc
  Pretrained model for molecular wavefunctions
  ☆49Updated 4 months ago
• n-gao / pesnet
  Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-f…
  ☆31Updated last year
• risi-kondor / GElib
  C++/CUDA library for SO(3) equivariant operations
  ☆25Updated last month
• mariogeiger / allegro-jax
  ☆22Updated 7 months ago
• mariogeiger / nequip-jax
  ☆23Updated 2 years ago
• metatensor / metatensor
  Self-describing sparse tensor data format for atomistic machine learning and beyond
  ☆90Updated this week
• atomicarchitects / nequix
  [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget.
  ☆52Updated 3 weeks ago
• e3nn / e3nn-jax
  jax library for E3 Equivariant Neural Networks
  ☆220Updated 3 months ago
• smsharma / eqnn-jax
  Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.
  ☆40Updated 9 months ago
• hatemhelal / mess
  MESS: Modern Electronic Structure Simulations
  ☆43Updated 2 months ago
• Luthaf / vesin
  Compute neighbor lists for atomistic systems
  ☆68Updated last month
• metatensor / featomic
  Computing representations for atomistic machine learning
  ☆74Updated this week
• mala-project / mala
  Materials Learning Algorithms. A framework for machine learning materials properties from first-principles data.
  ☆97Updated 2 months ago