Skip to content

gwatkinson/graphium

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

The Graph Of LIfe Library.


Binder PyPI Conda PyPI - Downloads Conda PyPI - Python Version license GitHub Repo stars GitHub Repo stars

A deep learning library focused on graph representation learning for real-world chemical tasks.

  • ✅ State-of-the-art GNN architectures.
  • 🐍 Extensible API: build your own GNN model and train it with ease.
  • ⚗️ Rich featurization: powerful and flexible built-in molecular featurization.
  • 🧠 Pretrained models: for fast and easy inference or transfer learning.
  • ⮔ Read-to-use training loop based on Pytorch Lightning.
  • 🔌 Have a new dataset? Goli provides a simple plug-and-play interface. Change the path, the name of the columns to predict, the atomic featurization, and you’re ready to play!

Try Online

Visit Binder and try Goli online.

Documentation

Visit https://valence-discovery.github.io/goli/.

Installation for developers

For CPU and GPU developers installs

Use either mamba or conda:

# Install mamba if unavailable
conda install -c conda-forge mamba

# Install Goli's dependencies in a new environment named `goli_dev`
mamba env create -f env.yml -n goli_dev

# Install Goli in dev mode
conda activate goli_dev
pip install -e .

For IPU developers installs

mkdir ~/.venv                           # Create the folder for the environment
python3 -m venv ~/.venv/goli_ipu        # Create the environment
source ~/.venv/goli_ipu/bin/activate    # Activate the environment

# Installing the poptorch SDK. Make sure to change the path
pip install PATH_TO_SDK/poptorch-3.2.0+109946_bb50ce43ab_ubuntu_20_04-cp38-cp38-linux_x86_64.whl

# Activate poplar SDK.
source PATH_TO_SDK/enable

# Install the IPU specific and goli requirements
PACKAGE_NAME=pytorch pip install -r requirements_ipu.txt

# Install Goli in dev mode
pip install -e .

Training a model

To learn how to train a model, we invite you to look at the documentation, or the jupyter notebooks available here.

If you are not familiar with PyTorch or PyTorch-Lightning, we highly recommend going through their tutorial first.

Changelogs

See the latest changelogs at CHANGELOG.rst.

License

Under the Apache-2.0 license. See LICENSE.

Authors

See AUTHORS.rst.

Documentation

  • Diagram for data processing in molGPS.

Data Processing Chart

  • Diagram for Muti-task network in molGPS

Full Graph Multi-task Network

About

Graphium: Scaling molecular GNNs to infinity.

Resources

License

Code of conduct

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 62.7%
  • Jupyter Notebook 37.2%
  • Shell 0.1%