Profile photo of Michael Jaemin Kim

About

I am a Research Scientist at Meta AI and Systems Codesign. I completed my PhD at Seoul National University, where I worked with Prof. Jung Ho Ahn, and was a visiting researcher at the University of Illinois Urbana-Champaign with Prof. Nam Sung Kim from March 2023 to March 2024.

My research focuses on improving the efficiency of GPUs for Large Language Model and Latent Diffusion Model inference, including power efficiency, higher-performance kernels, and KV cache offloading and prefix caching to CPU memory, disaggregated and CXL memory, and SSD for agentic and multi-turn inference. I am also interested in maximizing sparsity in model weights and activations, as well as improving HBM architecture and the memory hierarchy of GPU systems.

Industry Projects

  • AMD MI350X LLM/LDM Inference Enablement and Optimization. Led enablement and optimization of LLM and LDM inference on AMD MI350X across multiple production models on vLLM and Gen2 stacks. Optimized attention kernels, Triton fused-MoE heuristics, kernel fusion, and auto-tuning configurations. Achieved 2x throughput over H100/MI300X at iso-latency for LLM, 2.14x on Flux diffusion, 3.5 to 4.3x on AVG lite, and 1.2x faster than B200 on Unigen-1.5-MoE.
  • AI-Assisted Iterative Kernel Optimization for MI350X. Developed an automated kernel optimization loop combining AI-guided search with hardware-specific tuning for LLM inference. Wrote CK tile RMSNorm achieving 94% peak memory bandwidth, FP8 fused MoE with 17% speedup via XCD tile remap, and offline-tuned AITER MoE kernels that were 15% faster at large batch sizes.
  • Power Roofline Analysis. Co-developed a power roofline methodology for GPU inference and studied energy-efficiency tradeoffs below max TDP for datacenter compute maximization under a power cap. Characterized power-bound decode attention kernels and explained kernel auto-tuning ranking flips from power excursion. Submitted to MICRO 2026.
  • Blackwell FP4 GEMM Root-Cause Analysis. Investigated GB300 versus GB200 FP4 GEMM regression and traced it to TMA load latency exposure: tensor core throughput scaled 1.5x while shared memory capacity and TMA latency remained unchanged, with long_scoreboard at 79 to 89%.
  • Delta Compression for GEMM Decomposition. Explored delta compression as lossless sparsification to reduce GEMM power. Best-case results yielded about 22% exact zeros, below the 75 to 90% threshold needed, while temporal delta on diffusion cross-attention K/V reached 97% zeros but spatial delta on weights reduced sparsity.
  • GPU Benchmarking Infrastructure. Built a unified GPU benchmarking repository with automated sweep support, power and clock telemetry, a time-series dashboard, and auto-tuning integration.

Experience

  • Research Scientist, Meta, AI and Systems Codesign, June 2025 to present
  • Visiting Researcher, University of Illinois Urbana-Champaign, Prof. Nam Sung Kim, March 2023 to March 2024
  • Republic of Korea Army, Mandatory military service, Korea Military Academy, November 2014 to August 2016

Education

  • PhD, Seoul National University, Prof. Jung Ho Ahn, August 2019 to 2025
  • B.S.E. in Electrical and Computer Engineering, Seoul National University, August 2019

Publications

  1. HPCA 2026 "RoMe: Row Granularity Access Memory System for Large Language Models".
    Hwayong Nam, Seungmin Baek, Jumin Kim, Michael Jaemin Kim, Jung Ho Ahn.
  2. HPCA 2026 "ReScue: Reliable and Secure CXL Memory".
    Chihun Song, Austin Antony Cruz, Michael Jaemin Kim, Minbok Wi, Gaohan Ye, Kyungsan Kim, Sangyeol Lee, Jung Ho Ahn, Nam Sung Kim.
  3. CAL 2025 "Per-Row Activation Counting on Real Hardware: Demystifying Performance Overheads". Selected for HPCA 2026 Best of CAL.
    Jumin Kim, Seungmin Baek, Minbok Wi, Hwayong Nam, Michael Jaemin Kim, Sukhan Lee, Kyomin Sohn, Jung Ho Ahn.
  4. S&P 2026 "SoK: Systematizing a Decade of Architectural RowHammer Defenses Through the Lens of Streaming Algorithms".
    Michael Jaemin Kim, Seungmin Baek, Jumin Kim, Hwayong Nam, Nam Sung Kim, Jung Ho Ahn.
  5. CAL 2025 "X-PPR: Post package repair for CXL memory".
    Chihun Song, Michael Jaemin Kim, Yan Sun, Houxiang Ji, Kyungsan Kim, TaeKyeong Ko, Jung Ho Ahn, Nam Sung Kim.
  6. ASPLOS 2025 "Marionette: A RowHammer Attack via Row Coupling".
    Seungmin Baek, Minbok Wi, Seonyong Park, Hwayong Nam, Michael Jaemin Kim, Nam Sung Kim, Jung Ho Ahn.
  7. ASPLOS 2025 "M5: Mastering page migration and memory management for CXL-based tiered memory systems".
    Yan Sun, Jongyul Kim, Douglas Yu, Jiyuan Zhang, Siyuan Chai, Michael Jaemin Kim, Hwayong Nam, Jaehyun Park, Eojin Na, Yifan Yuan, Ren Wang, Jung Ho Ahn, Tianyin Xu, Nam Sung Kim.
  8. IEEE ESL 2024 "Hechi: A Hybrid Approach for Efficient Memory Reclamation Techniques in Mobile Systems".
    Wanju Doh, Seoyoung Ko, Michael Jaemin Kim, Jung Ho Ahn.
  9. ISCA 2024 "DRAMScope: Uncovering DRAM Microarchitecture and Characteristics by Issuing Memory Commands".
    Hwayong Nam, Seungmin Baek, Minbok Wi, Michael Jaemin Kim, Jaehyun Park, Chihun Song, Nam Sung Kim, Jung Ho Ahn.
  10. ASPLOS 2024 "AttAcc! Unleashing the Power of PIM for Batched Transformer-based Generative Model Inference".
    Jaehyun Park, Jaewan Choi, Kwanhee Kyung, Michael Jaemin Kim, Yongsuk Kwon, Nam Sung Kim, Jung Ho Ahn.
  11. ASPLOS 2024 "TAROT: A CXL SmartNIC-Based Defense Against Multi-bit Errors by Row-Hammer Attacks".
    Chihun Song, Michael Jaemin Kim, Tianchen Wang, Houxiang Ji, Jinghan Huang, Ipoom Jeong, Jaehyun Park, Hwayong Nam, Minbok Wi, Jung Ho Ahn, Nam Sung Kim.
  12. MICRO 2023 "How to Kill the Second Bird with One ECC: The Pursuit of Rowhammer Resilient DRAM".
    Michael Jaemin Kim, Jaehyun Park, Minbok Wi, Seoyoung Ko, Jaeyoung Choi, Hwayong Nam, Nam Sung Kim, Eojin Lee, Jung Ho Ahn.
  13. CAL 2023 "X-ray: Discovering DRAM internal structure and error characteristics by issuing memory commands".
    Hwayong Nam, Seunngmin Baek, Minbok Wi, Michael Jaemin Kim, Jaehyun Park, Chihun Song, Nam Sung Kim, Jung Ho Ahn.
  14. HPCA 2023 "SHADOW: Preventing Row Hammer in DRAM with Intra-Subarray Row Shuffling".
    Minbok Wi, Jaehyun Park, Michael Jaemin Kim, Seoyoung Ko, Nam Sung Kim, Eojin Lee, Jung Ho Ahn.
  15. arXiv "AESPA: Accuracy Preserving Low-degree Polynomial Activation for Fast Private Inference".
    Jaehyun Park, Michael Jaemin Kim, Wonkyung Jung, Jung Ho Ahn.
  16. IEEE TC 2022 "Future Scaling of Memory Hierarchy for Tensor Cores and Eliminating Redundant Shared Memory Traffic Using Inter-Warp Multicasting".
    Sunjung Lee, Seunghwan Hwang, Michael Jaemin Kim, Jaewan Choi, Jung Ho Ahn.
  17. ISCA 2022 "BTS: An Accelerator for Bootstrappable Fully Homomorphic Encryption".
    Sangpyo Kim, Jongmin Kim, Michael Jaemin Kim, Wonkyung Jung, John Kim, Minsoo Rhu, Jung Ho Ahn.
  18. HPCA 2022 "Mithril: Cooperative Row Hammer Protection on Commodity DRAM Leveraging Managed Refresh".
    Michael Jaemin Kim, Jaehyun Park, Yeonhong Park, Wanju Doh, Namhoon Kim, Tae Jun Ham, Jae W. Lee, Jung Ho Ahn.

Service

  • MICRO regular PC member, 2026
  • IEEE Transactions on Computers reviewer, 2024
  • IEEE Computer Architecture Letters reviewer, 2025 to 2026