Using the code for 3D data

[Nd-sole]

Hi @kevinzakka, thank you for making your well written code public. I am trying to use DRAM for the 3D data (videos). Can you advise me if your code can be extended to 3D data?