n-gao / pesnet
Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-free Inference ob Ab-Initio Potential Energy Surface Networks" (ICLR 2023)
☆30Updated 9 months ago
Alternatives and similar repositories for pesnet:
Users that are interested in pesnet are comparing it to the libraries listed below
- DeepErwin is a python 3.8+ package that implements and optimizes JAX 2.x wave function models for numerical solutions to the multi-electr…☆54Updated 3 months ago
- Supporting code for "Autoregressive neural-network wavefunctions for ab initio quantum chemistry".☆41Updated 2 years ago
- ESI-DCAFM-TACO-VDSP Summer School on "Machine Learning for Materials Hard and Soft"☆36Updated 2 years ago
- Equivariant machine learning interatomic potentials in JAX.☆71Updated last year
- PySCF with auto-differentiation☆77Updated this week
- Code repository for "Finding symmetry breaking order parameters with Euclidean Neural Networks"☆14Updated 4 years ago
- Code for performing adversarial attacks on atomistic systems using NN potentials☆36Updated 2 years ago
- variational free-energy of dense hydrogen☆12Updated last year
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆46Updated last year
- Differentiable Quantum Chemistry (only Differentiable Density Functional Theory and Hartree Fock at the moment)☆109Updated 3 years ago
- ☆40Updated 3 years ago
- Higher order equivariant graph neural networks for 3D point clouds☆36Updated last year
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆24Updated 11 months ago
- Tools for building equivariant polynomials on reductive Lie groups.☆30Updated last year
- Huxel: Huckel model + JAX (parameter optimization)☆10Updated 2 years ago
- Higher-order equivariant neural networks for charge density prediction in materials☆50Updated 3 weeks ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆103Updated 4 months ago
- Arbitrary-order derivatives of popular electronic structure methods, such as Hartree-Fock and coupled cluster theory.☆63Updated 6 months ago
- Flexible, general-purpose VMC framework, built on JAX.☆26Updated last week
- GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine l…☆84Updated last year
- GemNet model in TensorFlow, as proposed in "GemNet: Universal Directional Graph Neural Networks for Molecules" (NeurIPS 2021)☆25Updated last year
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆13Updated 3 weeks ago
- Exchange correlation functionals translated from libxc to jax☆43Updated last year
- MESS: Modern Electronic Structure Simulations☆19Updated 5 months ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆18Updated 3 weeks ago
- Build neural networks for machine learning force fields with JAX☆110Updated last month
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆30Updated 2 weeks ago
- Steerable E(3) GNN in jax☆24Updated last year
- [TMLR 2024] Generalizing Denoising to Non-Equilibrium Structures Improves Equivariant Force Fields☆31Updated last month
- Forward mode laplacian implemented in JAX tracer☆29Updated 2 weeks ago