Mentions légales du service

Skip to content
Snippets Groups Projects
  1. Apr 26, 2023
  2. Apr 24, 2023
  3. Apr 17, 2023
  4. Apr 06, 2023
  5. Apr 05, 2023
    • ANDREY Paul's avatar
      Refactor 'TorchModel' backend code to compute sample-wise gradients. · be449f94
      ANDREY Paul authored
      * Cache the functorch-built vmapped functions that compute (and opt.
        clip) sample-wise gradients.
      * When available, use `functorch.compile` to try and optimize these
        functions (resulting in lower execution runtimes on repeated calls
        thanks to their caching).
      * Note: with the release of torch-2.0, the `functorch.compile` API,
        which is experimental as of torch-1.13, is already deprecated.
        Hence this patch is going to be short-lived - but it may in turn
        pave the way to the future backend changes to add support for and
        take advantage of new torch-2.0 features.
      Verified
      be449f94
  6. Mar 29, 2023
  7. Mar 28, 2023
Loading