Refactor to use tyro #1477

Job	Run time
test-atari-envs (3.8, 1.3.1, ubuntu-22.04)	6m 39s
test-atari-multigpu-envs (3.8, 1.3.1, ubuntu-22.04)	3m 8s
test-atari-envs (3.8, 1.3.1, macos-latest)	9m 36s
test-atari-envs (3.8, 1.3.1, windows-latest)	8m 9s
test-core-envs (3.8, 1.3.1, ubuntu-22.04)	5m 2s
test-core-envs (3.8, 1.3.1, macos-latest)	6m 37s
test-envpool-envs (3.8, 1.3.1, ubuntu-22.04)	5m 23s
test-core-envs (3.8, 1.3.1, windows-latest)	5m 46s
test-mujoco_py-envs (3.8, 1.3.1, ubuntu-22.04)	5m 3s
test-mujoco-envs (3.8, 1.3.1, ubuntu-22.04)	4m 40s
test-procgen-envs (3.8, 1.3.1, ubuntu-22.04)	3m 13s
test-procgen-envs (3.8, 1.3.1, macos-latest)	3m 9s
test-procgen-envs (3.8, 1.3.1, windows-latest)	5m 11s
test-pettingzoo-envs (3.8, 1.3.1, ubuntu-22.04)	3m 4s
	1h 14m 40s

Provide feedback