SSealbook
Sealbook / Open source / Machine learning & data science

gordicaleksa/pytorch-GAT

My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!

attentionattention-mechanismdeep-learninggatgat-tutorialgraph-attention-networkgraph-attention-networksjupyterpythonpytorch
GitHub ↗Website ↗

License & maintenance

MIT is a permissive license: free to use commercially with attribution.

Last code push 3 years ago.

Similar projects

RepositoryStarsLanguageLicenseActivity
1labmlai/annotated_deep_learning_paper_implementations🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback,★ 67.6kPythonMITSlowing
2keras-team/kerasDeep Learning for humans★ 64.4kPythonApache-2.0Active
3ultralytics/ultralyticsUltralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose esti★ 62.6kPythonAGPL-3.0Active
4CorentinJ/Real-Time-Voice-CloningClone a voice in 5 seconds to generate arbitrary speech in real-time★ 60.2kPythonOtherSlowing
5coqui-ai/TTS🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production★ 46.1kPythonMPL-2.0Stale
6Lightning-AI/pytorch-lightningPretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.★ 31.4kPythonApache-2.0Active
7fastai/fastaiThe fastai deep learning library★ 28.2kJupyter NotebookApache-2.0Active
8ritchieng/the-incredible-pytorchThe Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relating to PyTorch. ★ 12.7k—MITActive