You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
This was brought up by @bertmaher offline, I am just writing an issue for it. It would be nice if we could have a set of models where JAX is more competitive than PyTorch and be able to run comparisons on both in torchbench.
This was brought up by @bertmaher offline, I am just writing an issue for it. It would be nice if we could have a set of models where JAX is more competitive than PyTorch and be able to run comparisons on both in torchbench.
Some prior work: https://github.com/TheSalon/fast-dpsgd benchmarked various differential private models between PT, JAX, and some other frameworks.
The text was updated successfully, but these errors were encountered: