ORNL / HydraGNN
Distributed PyTorch implementation of multi-headed graph convolutional neural networks
☆78Updated last week
Alternatives and similar repositories for HydraGNN:
Users that are interested in HydraGNN are comparing it to the libraries listed below
- python library for atomistic machine learning☆76Updated last week
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆49Updated this week
- C++/CUDA library for SO(3) equivariant operations☆23Updated this week
- Open MatSci ML Toolkit is a framework for prototyping and scaling out deep learning models for materials discovery supporting widely used…☆171Updated 3 weeks ago
- Workflow for creating and analyzing the Open Catalyst Dataset☆106Updated 2 months ago
- Equivariant machine learning interatomic potentials in JAX.☆72Updated last year
- Computing representations for atomistic machine learning☆70Updated last month
- ☆44Updated 10 months ago
- Generate and predict molecular electron densities with Euclidean Neural Networks☆47Updated last year
- This is a repository with examples to run inference endpoints on various ALCF clusters☆19Updated last week
- Higher-order equivariant neural networks for charge density prediction in materials☆54Updated last month
- Jax / Haiku implementation of DimeNet++.☆18Updated 3 years ago
- Reference implementation of "SpookyNet: Learning force fields with electronic degrees of freedom and nonlocal effects"☆72Updated 2 years ago
- Higher order equivariant graph neural networks for 3D point clouds☆36Updated last year
- Repository to host supporting information and code samples for Accelerated DFT☆23Updated 6 months ago
- A reinforcement learning library for material and molecule optimization☆30Updated last year
- Official implementation of DeepDFT model☆75Updated 2 years ago
- A many-body extension of the FLARE code.☆36Updated 3 years ago
- A scalable and versatile library to generate representations for atomic-scale learning☆80Updated last year
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆103Updated last week
- MESS: Modern Electronic Structure Simulations☆19Updated 6 months ago
- ☆23Updated last year
- GemNet model in PyTorch, as proposed in "GemNet: Universal Directional Graph Neural Networks for Molecules" (NeurIPS 2021)☆196Updated last year
- These are the slides associated with the GNN tutorial at the APS March Meeting☆21Updated 2 years ago
- Implementation of a machine learned density functional☆36Updated 10 months ago
- This repository is a package to provide SAIT Machine Learning Force Field(MLFF) Framework☆35Updated last year
- 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
- Predict the electronic structure and atomic properties (potential energy, forces, and stress tensor) of polymers containing N and/or O.☆19Updated 7 months ago
- A JAX library for Density Functional Theory.☆53Updated last month
- GauXC is a modern, modular C++ library for the evaluation of quantities related to the exchange-correlation (XC) energy (e.g. potential, …☆33Updated 2 weeks ago