Klas Wijk
PhD Student at KTH Royal Institute of Technology
Lindstedtsvägen 24
Stockholm, Sweden
About
I am a PhD student in Computer Science at KTH Royal Institute of Technology, supervised by Hossein Azizpour, and co-supervised by Ricardo Vinuesa at the University of Michigan. My expected graduation date is in late 2027. Before starting my PhD, I received a BSc in Computer Science and an MSc in Applied and Computational Mathematics at KTH.
My research is focused on different aspects of top-k: relaxations, sampling, and gradient estimation. I have also worked on generative models and inverse problems in fluid mechanics as part of a collaborative project. More generally, I am interested in all things machine learning, statistics, and applied mathematics.
My PhD is funded by the Swedish e-Science Research Centre (SeRC). I am also part of the Wallenberg AI, Autonomous Systems and Software Program (WASP), the largest research program in Sweden.
News
| Jan 14, 2026 | Attended the yearly WASP Winter Conference and presented a poster. |
|---|---|
| Oct 31, 2025 | Poster accepted to the EurIPS’25 DiffSys Workshop in Copenhagen! 🇩🇰 |
| Oct 14, 2025 | Poster accepted to the ELLIS UnConference in Copenhagen! 🇩🇰 |
| Apr 28, 2025 | Attended ICLR 2025 in Singapore and presented a poster! 🇸🇬 |
| Jan 22, 2025 | Paper accepted to ICLR 2025. |
Latest posts
| Jan 30, 2026 | Top-k Sampling Beyond Gumbel Top-k |
|---|
