Skip to content

Commit cfc5916

Browse files
committed
rollout notebook: fixups to benchmarks, plots, text
1. MJX benchmarks use jax.lax.scan and block_until_ready 2. Benchmark parameters updated to match RL use case 3. humanoid benchmarks are warmstarted by 2 seconds, humanoid100 by 4 seconds 4. benchmarks use skip_checks and preallocated threadpools 5. Improved advanced usage plots 6. Benchmark no longer includes model initialization in timed section 7. Update notebook text as needed
1 parent 14c8dc6 commit cfc5916

File tree

1 file changed

+223
-159
lines changed

1 file changed

+223
-159
lines changed

0 commit comments

Comments
 (0)