Sparse MatMul Support in TorchJD for Seamless compatibility with torch.vmap() #1254
| Job | Run time |
|---|---|
| 44s | |
| 1m 6s | |
| 41s | |
| 1m 6s | |
| 2m 19s | |
| 1m 12s | |
| 2m 24s | |
| 49s | |
| 47s | |
| 1m 10s | |
| 2m 59s | |
| 42s | |
| 36s | |
| 1m 6s | |
| 2m 23s | |
| 20m 4s |
| Job | Run time |
|---|---|
| 44s | |
| 1m 6s | |
| 41s | |
| 1m 6s | |
| 2m 19s | |
| 1m 12s | |
| 2m 24s | |
| 49s | |
| 47s | |
| 1m 10s | |
| 2m 59s | |
| 42s | |
| 36s | |
| 1m 6s | |
| 2m 23s | |
| 20m 4s |