Skip to content

Credits

Author

Aleksandra Ciprijanovic GitHub: @AleksCipri

ShiftKit is a modular, science-focused domain adaptation framework for PyTorch, built around clean interfaces for fast implementation of different deep learning architectures, datasets, and domain adaptation methods.


Other Contributors

Karla Tame-Narvaez GitHub: @karlaTame

Abdelrahman Helal GitHub: @abdelrahman-helal

Added PyTorch Geometric support: GNN model, node-level and graph-level domain adaptation, PyG data utilities, and the node-level MMD example.


Dependencies

ShiftKit builds on the following open-source libraries:

Library Version Use
PyTorch ≥ 2.0 Deep learning backend
torchvision ≥ 0.15 Built-in datasets (MNIST)
NumPy ≥ 1.24 Numerical operations
scikit-learn ≥ 1.2 t-SNE for latent space visualisation
matplotlib ≥ 3.7 All plotting and diagnostics
tqdm ≥ 4.65 Training progress bars

Optional dependencies (required only for specific methods):

Library Version Use
torch-geometric ≥ 2.0 GNN models and PyG graph data (shiftkit.models.GNN)
geomloss ≥ 0.2 Sinkhorn divergence for SIDDATrainer

Documentation built with MkDocs and the Material for MkDocs theme.


Licence

ShiftKit is released under the Apache License 2.0.