Amazing, @junpenglao! Right now we are stuck documenting current results in a paper. But as soon as I finish this part, I will try to contact you. Actually multi-batch would be very useful since we analyze and simulate for several locations. TFP looks pretty exciting, I have to investigate it further.