jln.shk_64x4: JSAE (resid_mid -> resid_out) 7.5k steps j.mlp_layer.shk_64x4: JSAE (mlp_in -> mlp_out) 7.5k steps