nats_bench>=1.4
pytorch>=1.5.0