An open-source AutoML Library based on PyTorch
copied from cf-staging / gradsflowGradsFlow is an open-source AutoML Library based on PyTorch. Our goal is to democratize AI and make it available to everyone.
It can automatically build & train Deep Learning Models for different tasks on your laptop or to a remote cluster directly from your laptop. It provides a powerful and easy-to-extend Model Training API that can be used to train almost any PyTorch model. Though GradsFlow has its own Model Training API it also supports PyTorch Lightning Flash to provide more rich features across different tasks.
