deepchem / jaxchem
JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling
☆79Updated 4 years ago
Alternatives and similar repositories for jaxchem:
Users that are interested in jaxchem are comparing it to the libraries listed below
- Differentiate all the things!☆145Updated this week
- Differentiable molecular simulation of proteins with a coarse-grained potential☆54Updated 2 weeks ago
- An experimental repo for experimenting with PyTorch models☆35Updated last year
- Library for training Gaussian Processes on Molecules☆36Updated 3 years ago
- ☆10Updated last year
- ☆48Updated 2 years ago
- Codebase for Cormorant Neural Networks☆60Updated 2 years ago
- A modular framework for neural networks with Euclidean symmetry☆11Updated 7 months ago
- Learning protein structure with a differentiable simulator☆27Updated 5 years ago
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks☆129Updated 3 years ago
- Graph Convolutional Networks in JAX☆31Updated 4 years ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆30Updated 2 weeks ago
- Arbitrary-order derivatives of popular electronic structure methods, such as Hartree-Fock and coupled cluster theory.☆62Updated 6 months ago
- ☆40Updated 3 years ago
- ☆86Updated 3 years ago
- Graph Inference on MoLEcular Topology☆26Updated last year
- ☆35Updated 5 years ago
- PySCF on IPU☆42Updated 11 months ago
- Reinforcement Learning for Molecular Design Guided by Quantum Mechanics☆125Updated last year
- parameterizing valid Euclidean distance matrices (EDMs) via neural networks☆19Updated 5 years ago
- ☆62Updated 5 years ago
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆25Updated 10 months ago
- Pytorch reimplementation of Molecule Attention Transformer, which uses a transformer to tackle the graph-like structure of molecules☆58Updated 4 years ago
- Steerable E(3) GNN in jax☆24Updated last year
- MaxEnt code for fitting simulation outcomes/statistical models to observations☆16Updated 2 years ago
- TLDRs for ML in Drug Discovery papers☆71Updated last year
- repository and website for tutorials on 3d Euclidean equivariant neural networks☆67Updated 3 years ago
- Data and models (with prediction scripts) used in publications related to Apollo1060 platform☆64Updated 8 months ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- Massively-Parallel Natural Extension of Reference Frame☆30Updated 2 years ago