OmniGibson icon indicating copy to clipboard operation
OmniGibson copied to clipboard

NumPy -> Torch

Open hang-yin opened this issue 7 months ago • 1 comments

High level changes:

  • Numpy -> Torch conversion
  • Enforce th.float32 whenever possible
  • Simulation backend -> torch
  • Rewrite transform utils with torch jit scripts
  • Use math instead of torch whenever possible
  • Fix robot state tests
  • Remove Scipy rotation library
  • Remove trimesh transformation library
  • Reenable examples as tests
  • Fix all examples (except attachment demo seg fault)

TODO:

  • [x] Bring performance back by getting rid of warp.synchronize() call when fetching data from annotators

hang-yin avatar Jun 27 '24 22:06 hang-yin