feature: add mps kernel for compute_puff_advantage
#422
+360
−11
compute_puff_advantage
#422