Performance improvements and neuron_parallel_compile and gradient checkpointing fixes
#491
| Job | Run time |
|---|---|
| 1h 5m 39s | |
| 1h 5m 39s |
neuron_parallel_compile and gradient checkpointing fixes
#491
| Job | Run time |
|---|---|
| 1h 5m 39s | |
| 1h 5m 39s |