remove torch from jd build