IntelLabs / matsciml

Open MatSci ML Toolkit is a framework for prototyping and scaling out deep learning models for materials discovery supporting widely used materials science datasets, and built on top of PyTorch Lightning, the Deep Graph Library, and PyTorch Geometric.
MIT License
144 stars 20 forks source link

Register Models and Tasks #250

Closed melo-gonzo closed 3 months ago

melo-gonzo commented 3 months ago

This PR ensures models and tasks are registered if they weren't already. Closes #111