Fix CI - Githubissues

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.

https://pytorch.org/functorch/

BSD 3-Clause "New" or "Revised" License

1.38k stars 102 forks source link

Closed samdow closed 1 year ago

samdow commented 1 year ago

kl div looks like it's passing on cuda and still failing on cpu so putting this as a PR to double check that