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
- Differentiable molecular simulation of proteins with a coarse-grained potential☆55Updated last month
- Differentiate all the things!☆149Updated this week
- An experimental repo for experimenting with PyTorch models☆35Updated 2 years ago
- Library for training Gaussian Processes on Molecules☆36Updated 3 years ago
- Learning protein structure with a differentiable simulator☆27Updated 5 years ago
- ☆50Updated 2 years ago
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks☆131Updated 3 years ago
- A modular framework for neural networks with Euclidean symmetry☆11Updated 10 months ago
- Graph Inference on MoLEcular Topology☆26Updated 2 years ago
- Graph Convolutional Networks in JAX☆32Updated 4 years ago
- ☆88Updated 3 years ago
- Massively-Parallel Natural Extension of Reference Frame☆31Updated 2 years ago
- ☆40Updated 3 years ago
- TLDRs for ML in Drug Discovery papers☆70Updated 2 years ago
- Codebase for Cormorant Neural Networks☆60Updated 2 years ago
- MaxEnt code for fitting simulation outcomes/statistical models to observations☆16Updated 2 years ago
- ☆35Updated 5 years ago
- 🔗 PyTorch implementation of the Parallelized Natural Extension Reference Frame algorithm☆19Updated 6 years ago
- An efficient method for the conversion from internal to Cartesian coordinates that utilizes the platform-agnostic JAX Python library.☆17Updated 10 months ago
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆24Updated last year
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- ☆10Updated 2 years ago
- Reinforcement Learning for Molecular Design Guided by Quantum Mechanics☆126Updated last year
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆32Updated 2 months ago
- Energy-based models for atomic-resolution protein conformations☆97Updated 3 years ago
- ☆28Updated 2 years ago
- Arbitrary-order derivatives of popular electronic structure methods, such as Hartree-Fock and coupled cluster theory.☆65Updated 8 months ago
- Universal Transforming Geometric Network for protein structure prediction.☆8Updated 4 years ago
- parameterizing valid Euclidean distance matrices (EDMs) via neural networks☆19Updated 5 years ago
- Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-f…☆30Updated 10 months ago