requirements.txt 156 Bytes
Newer Older
jerrrrry's avatar
jerrrrry committed
1
2
3
4
5
6
einops
zarr
tensorstore!=0.1.46,!=0.1.72
torch
nvidia-modelopt[torch]>=0.19.0; sys_platform != "darwin"
nvidia-resiliency-ext; platform_machine == "x86_64"