DeepChem is part of Open Chemistry, which will be applying for Google summer of code this summer. We don’t yet know if we will be selected, but I wanted to start a thread up to brainstorm ideas for potential student projects:
JaxModel: DeepChem currently doesn’t have a good way to build models with Jax. This project would work to add a wrapper
JaxModelin the style of
TorchModelthat allows for convenient wrapping of arbitrary Jax models in DeepChem. The project will involve implementing
JaxModel, writing a suitable test suite, and putting together a good tutorial on how to use Jax with DeepChem as a jupyter notebook
- Pytorch Lightning support: PyTorch lightning is a popular framework for PyTorch. This project would look into enabling the easy construction of PyTorch lightning based models for DeepChem
- Paper Implementation: This project is more open ended. Pick a research paper that you like and implement it within DeepChem. For success in this project, you should reach out to me or other DeepChem community members for feedback and help in picking a suitable project.
If you have other suggestions for GSoC projects, please post them here!