per-sample-grad performance comparison with pytorch/opacus #624

Open

Open

per-sample-grad performance comparison with pytorch/opacus#624

Labels

We should check performance of vmap x grad on pytorch/opacus' examples:

https://github.com/pytorch/opacus/tree/main/examples

Metadata

Assignees

No one assigned

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests