ORNL / HydraGNN
Distributed PyTorch implementation of multi-headed graph convolutional neural networks
☆79Updated 2 weeks ago
Alternatives and similar repositories for HydraGNN:
Users that are interested in HydraGNN are comparing it to the libraries listed below
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆57Updated this week
- Open MatSci ML Toolkit is a framework for prototyping and scaling out deep learning models for materials discovery supporting widely used…☆173Updated last month
- Self-describing sparse tensor data format for atomistic machine learning and beyond☆73Updated this week
- python library for atomistic machine learning☆77Updated last week
- Higher order equivariant graph neural networks for 3D point clouds☆37Updated last year
- Equivariant machine learning interatomic potentials in JAX.☆72Updated 2 weeks ago
- Workflow for creating and analyzing the Open Catalyst Dataset☆107Updated 3 months ago
- Jax / Haiku implementation of DimeNet++.☆18Updated 3 years ago
- ☆37Updated 9 months ago
- C++/CUDA library for SO(3) equivariant operations☆23Updated 2 weeks ago
- cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffD…☆214Updated this week
- GemNet model in PyTorch, as proposed in "GemNet: Universal Directional Graph Neural Networks for Molecules" (NeurIPS 2021)☆200Updated 2 years ago
- Deprecated - see `pair_nequip_allegro`☆43Updated 2 weeks ago
- Computing representations for atomistic machine learning☆71Updated last week
- Implementation of a machine learned density functional☆36Updated 10 months ago
- Data and code for graph neural network accelerated molecular dynamics☆40Updated 3 years ago
- A JAX library for Density Functional Theory.☆53Updated last week
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆103Updated last month
- Generate and predict molecular electron densities with Euclidean Neural Networks☆47Updated last year
- Reference implementation of "SpookyNet: Learning force fields with electronic degrees of freedom and nonlocal effects"☆72Updated 3 years ago
- These are the slides associated with the GNN tutorial at the APS March Meeting☆21Updated 2 years ago
- Official implementation of DeepDFT model☆75Updated 2 years ago
- Matbench: Benchmarks for materials science property prediction☆155Updated 8 months ago
- ☆23Updated last year
- An overview of literature that discusses the use of machine learning for atomistic simulations☆45Updated 2 years ago
- Build neural networks for machine learning force fields with JAX☆115Updated this week
- Learning neural network potentials from experimental data via Differentiable Trajectory Reweighting☆31Updated last year
- Higher-order equivariant neural networks for charge density prediction in materials☆56Updated 2 months ago
- Differentiable Quantum Chemistry (only Differentiable Density Functional Theory and Hartree Fock at the moment)☆111Updated 3 years ago
- Keras layers for end-to-end learning with rdkit and pymatgen☆60Updated last year