Thanks for the updates. Could you also post the mini-batch version of fastGTN and GTN as mentioned in the paper? They would be very helpful.