equi2cube TODO: Implement numpy Implement torch Implement torch with batches Fix rotation axis Implement c++ with cuda