Git Product home page Git Product logo

openchem's Introduction

OpenChem

OpenChem

OpenChem is a deep learning toolkit for Computational Chemistry with PyTorch backend. The goal of OpenChem is to make Deep Learning models an easy-to-use tool for Computational Chemistry and Drug Design Researchers.

Main features

  • Modular design with unified API, modules can be easily combined with each other.
  • OpenChem is easy-to-use: new models are built with only configuration file.
  • Fast training with multi-gpu support.
  • Utilities for data preprocessing.
  • Tensorboard support.

Documentation

Check out OpenChem documentation here.

Supported functionality

Tasks:

  • Classification (binary or multi-class)
  • Regression
  • Multi-task (such as N binary classification tasks)
  • Generative models

Data types

  • Sequences of characters such as SMILES strings or amino-acid sequences
  • Molecular graphs. OpenChem takes care of converting SMILES strings into molecular graphs

Modules:

  • Token embeddings
  • Recurrent neural network encoders
  • Graph convolution neural network encoders
  • Multi-layer perceptrons

We are working on populating OpenChem with more models and other building blocks.

Installation

Requirements

In order to get started you need:

General installation

If you installed your Python with Anaconda you can run the following commands to get started:

git clone https://github.com/Mariewelt/OpenChem.git
cd OpenChem
conda create --name OpenChem python=3.7
conda activate OpenChem
conda install --yes --file requirements.txt
conda install -c rdkit rdkit nox cairo
conda install pytorch torchvision -c pytorch
pip install -e .

If your CUDA version is older than 9.0, check Pytorch website for different installation instructions.

Installation with Docker

Alternative way of installation is with Docker. We provide a Dockerfile, so you can run your models in a container that already has all the necessary packages installed. You will also need nvidia-docker in order to run models on GPU.

Publications

If you use OpenChem in your projects, please cite:

Korshunova, Maria, et al. "OpenChem: A Deep Learning Toolkit for Computational Chemistry and Drug Design." Journal of Chemical Information and Modeling 61.1 (2021): 7-13.

MolecularRNN model paper:

Popova, Mariya, et al. "MolecularRNN: Generating realistic molecular graphs with optimized properties." arXiv preprint arXiv:1905.13372 (2019).

Acknowledgements

OpenChem was supported by Carnegie Mellon University, the University of North Carolina at Chapel Hill and NVIDIA Corp.

CMU
UNC NVIDIA

openchem's People

Contributors

mariewelt avatar itisalirh avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.