Closed vmoens closed 1 month ago
@jisaacso this is the last PR of the compile stack
It includes benchmarks for tensordict vs pytree in eager and compile. Ops like copy
(test: test_compile_copy_flat
) run faster in eager and compile with TD compared to pytree.
Stack from ghstack (oldest at bottom):