batching

thank you for the amazing code!

just got it training today in colab. i would like to try implementing batching. i saw it listed in the engineering TODOs. did you try this at all yet? anything to watch out for?

thanks again for making splatting more accessible!