Update README.md

README.md

@@ -2,14 +2,17 @@
 license: mit
 ---
 
-# Walrus Steering Checkpoint
+# Physics Steering: Causal Control of Cross-Domain Concepts in a Physics Foundation Model
 
-This checkpoint contains weights for the [Walrus](https://huggingface.co/polymathic-ai/walrus) physics foundation model, fine-tuned and used for activation steering experiments described in:
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![GitHub Repo](https://img.shields.io/badge/GitHub-Repo-blue?logo=github)](https://github.com/DJ-Fear/walrus_steering)
+[![arXiv](https://img.shields.io/badge/arXiv-2511.20798-b31b1b.svg)](https://arxiv.org/abs/2511.20798)
 
-> **Physics Steering: Causal Control of Cross-Domain Concepts in a Physics Foundation Model** — [arXiv:2511.20798v2](https://arxiv.org/abs/2511.20798)
+This checkpoint contains weights for the [Walrus](https://huggingface.co/polymathic-ai/walrus) physics foundation model, fine-tuned and used for activation steering experiments described in *Physics Steering: Causal Control of Cross-Domain Concepts in a Physics Foundation Model* — [arXiv:2511.20798v2](https://arxiv.org/abs/2511.20798).
 
 The base Walrus model is a 1.3B-parameter space–time Transformer trained on 19 physical simulation datasets. See the main [Walrus repo](https://github.com/PolymathicAI/walrus) for full details on the architecture and pretraining.
 
+
 ## Usage
 
 Download the checkpoint and point your config at it: