lit-learn: A Toolkit for Deep Learning

lit-learn is a Python toolkit designed for deep learning built on PyTorch Lightning and Lightning Fabric.

📦 Installation

git clone https://github.com/tanganke/lit-learn.git
cd lit-learn
pip install -e .

📚 Documentation

Comprehensive documentation is available:

cd docs
make html

🎯 Roadmap

• Core architecture with Lightning/Fabric support
• Multi-task learning algorithms
• Multi-objective optimization methods
• Advanced Pareto optimization (NSGA-II, MOGA)
• Bayesian optimization integration
• AutoML for hyperparameter tuning
• More pre-built algorithms
• Integration with popular datasets

🤝 Contributing

Contributions are welcome! Please feel free to submit a Pull Request. For major changes, please open an issue first to discuss what you would like to change.

📄 License

This project is licensed under the MIT License - see the LICENSE file for details.