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 2 weeks 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
- GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine l…☆100Updated last year
- PySCF with auto-differentiation☆79Updated 3 weeks ago
- variational free-energy of dense hydrogen☆12Updated last year
- Calculate observables from neural network-based VMC (NN-VMC).☆15Updated 3 weeks ago
- Code repository for "Finding symmetry breaking order parameters with Euclidean Neural Networks"☆14Updated 4 years ago
- Equivariant machine learning interatomic potentials in JAX.☆72Updated last week
- Exchange correlation functionals translated from libxc to jax☆45Updated last month
- Arbitrary-order derivatives of popular electronic structure methods, such as Hartree-Fock and coupled cluster theory.☆65Updated 8 months ago
- Differentiable Quantum Chemistry (only Differentiable Density Functional Theory and Hartree Fock at the moment)☆111Updated 3 years ago
- Higher order equivariant graph neural networks for 3D point clouds☆37Updated last year
- A JAX library for Density Functional Theory.☆53Updated last week
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆32Updated 2 months ago
- MESS: Modern Electronic Structure Simulations☆20Updated 7 months ago
- Steerable E(3) GNN in jax☆24Updated last year
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆46Updated last year
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆103Updated last month
- MESS: Modern Electronic Structure Simulations☆28Updated last month
- Flexible, general-purpose VMC framework, built on JAX.☆26Updated 2 weeks ago
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆24Updated last year
- Code for performing adversarial attacks on atomistic systems using NN potentials☆36Updated 2 years ago
- Pytorch Implementation of Real Space Quantum Monte Carlo Simulations of Molecular Systems☆27Updated this week
- Huxel: Huckel model + JAX (parameter optimization)☆10Updated 2 years ago
- Python library for real space quantum Monte Carlo☆87Updated this week
- Nomalizing flows for orbita-free DFT☆10Updated 7 months ago
- JAX accelerated Quantum Monte Carlo☆75Updated last month
- Forward mode laplacian implemented in JAX tracer☆29Updated 2 months ago
- Build neural networks for machine learning force fields with JAX☆115Updated this week