ai2thor>=5.0.0
flax>=0.12.0
hydra-core>=1.3.2
ipdb>=0.13.13
matplotlib>=3.9.0
moviepy>=2.2.1
natsort>=8.4.0
networkx>=3.5
open3d>=0.19.0
openai>=2.0.0
polars>=1.33.1
prior>=1.0.3
scienceplots>=2.1.1
seaborn>=0.13.2
shapely>=2.1.2
tenacity>=9.1.2
tiktoken>=0.11.0
wheel>=0.45.1

[:sys_platform == "darwin"]
jax-metal>=0.1.1

[:sys_platform == "linux"]
jax[cuda12]>=0.8.1
