Kai-Jie Lin (KJ) 林楷傑

I am an M.S. student in Electrical and Computer Engineering at Carnegie Mellon University. I am working on RL-Kernel GitHub stars, an open-source library of bitwise-consistent CUDA/Triton kernels for RL post-training. My research interests are Reinforcement Learning, Large Reasoning Models, and Machine Learning Systems.

Previously, I was an Research Assistant in the Reinforcement Learning and Bandits Lab at NYCU and an Machine Learning Engineer at Rayleigh Vision Intelligence, building multimodal machine learning systems for production lines. I received my B.S. in Computer Science from National Yang Ming Chiao Tung University, advised by Prof. Ping-Chun Hsieh. In fall 2024 I was an exchange student at the University of Illinois Urbana-Champaign.

Kai-Jie Lin (KJ) profile photo

News

Publications

BOFormer: Learning to Solve Multi-Objective Bayesian Optimization via Non-Markovian RL
BOFormer: Learning to Solve Multi-Objective Bayesian Optimization via Non-Markovian RL
Kai-Jie Lin, Yu Heng Hung, Yu-Heng Lin, Chien-Yi Wang, Cheng Sun, Ping-Chun Hsieh
International Conference on Learning Representations (ICLR), 2025

We propose BOFormer, a generalized deep Q-learning framework for Multi-Objective Bayesian Optimization (MOBO) via sequence modeling.