Add pipeline tag and paper link

README.md

@@ -1,18 +1,19 @@
 ---
-license: mit
+datasets:
+- mlfoundations/dclm-baseline-1.0
 library_name: megatron-lm
+license: mit
 tags:
-  - mixture-of-experts
-  - moe
-  - distributionally-robust-optimization
-  - megatron
-datasets:
-  - mlfoundations/dclm-baseline-1.0
+- mixture-of-experts
+- moe
+- distributionally-robust-optimization
+- megatron
+pipeline_tag: text-generation
 ---
 
 # DRMoET checkpoints
 
-<!-- TODO: paper title, arXiv link, authors -->
+**Paper:** [Distributionally Robust Mixture-of-Experts Training](https://huggingface.co/papers/2610.07207)
 
 **Project page:** <https://drmoet.github.io> **Code:** <https://github.com/MAPS-research/DRMoET>