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 10 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
- ESI-DCAFM-TACO-VDSP Summer School on "Machine Learning for Materials Hard and Soft"☆36Updated 2 years ago
- Supporting code for "Autoregressive neural-network wavefunctions for ab initio quantum chemistry".☆41Updated 2 years ago
- Equivariant machine learning interatomic potentials in JAX.☆72Updated last year
- Code repository for "Finding symmetry breaking order parameters with Euclidean Neural Networks"☆14Updated 4 years ago
- Differentiable Quantum Chemistry (only Differentiable Density Functional Theory and Hartree Fock at the moment)☆110Updated 3 years ago
- GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine l…☆94Updated last year
- PySCF with auto-differentiation☆79Updated this week
- Arbitrary-order derivatives of popular electronic structure methods, such as Hartree-Fock and coupled cluster theory.☆65Updated 7 months ago
- Huxel: Huckel model + JAX (parameter optimization)☆10Updated 2 years ago
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆46Updated last year
- Higher order equivariant graph neural networks for 3D point clouds☆36Updated last year
- variational free-energy of dense hydrogen☆12Updated last year
- A JAX library for Density Functional Theory.☆53Updated last month
- Code for performing adversarial attacks on atomistic systems using NN potentials☆36Updated 2 years ago
- Exchange correlation functionals translated from libxc to jax☆45Updated 3 weeks ago
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆103Updated last week
- JAX accelerated Quantum Monte Carlo☆74Updated last month
- MESS: Modern Electronic Structure Simulations☆19Updated 6 months ago
- Higher-order equivariant neural networks for charge density prediction in materials☆54Updated last month
- Flexible, general-purpose VMC framework, built on JAX.☆26Updated 3 weeks ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆31Updated last month
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆24Updated last year
- ☆11Updated last year
- GemNet model in TensorFlow, as proposed in "GemNet: Universal Directional Graph Neural Networks for Molecules" (NeurIPS 2021)☆25Updated last year
- Corresponding dataset and tools for the AdsorbML manuscript.☆40Updated 2 months ago
- Pytorch Implementation of Real Space Quantum Monte Carlo Simulations of Molecular Systems☆26Updated this week
- [TMLR 2024] Generalizing Denoising to Non-Equilibrium Structures Improves Equivariant Force Fields☆32Updated 2 months ago
- Computing representations for atomistic machine learning☆70Updated last month
- an interface to semi-empirical quantum chemistry methods implemented with pytorch☆49Updated 2 weeks ago