Add pipeline tag and project/code links

README.md

@@ -1,12 +1,17 @@
 ---
-license: apache-2.0
 datasets:
 - HuggingFaceFW/fineweb-edu
+license: apache-2.0
+pipeline_tag: text-generation
 ---
+
 Checkpoints evaluated in the paper "Graph Machine: Towards Better Pretraining via Edges" (https://arxiv.org/abs/2609.02881):
 
 We introduce the Graph Machine (GM), an architecture that maintains an O(n)-sized state and accesses it through sparse, dynamic routing.
 Unlike methods with fixed-size states or sparse but static routing, GM preserves O(n) complexity in its sparse layers without restricting the potentially accessible state size to O(1).
 Instead, GM uses edges - pointer-like objects updated differentiably by a referral mechanism resembling pointer chasing.
 We replace 75% of the dense Transformer layers in Qwen3-0.6B with GM sparse layers and pretrain from scratch on 15.7B tokens.
-With only 2 of 4,096 tokens retrieved per KV head in each sparse layer, loss degrades only slightly; with 4, the best model marginally improves loss.
+With only 2 of 4,096 tokens retrieved per KV head in each sparse layer, loss degrades only slightly; with 4, the best model marginally improves loss.
+
+Project page: https://lintaihou.com/research/graph-machine-2  
+Code: https://github.com/lintaihou/gm2