torch
torchvision
cvxpy
open3d
wandb
einops
fpsample
gym
ninja
