Ben Snyder

icon indicating a website link https://www.linkedin.com/in/johnbensnyder/ icon indicating an email [email protected]

icon company Amazon Web Services icon location Palo Alto, CA icon location Machine learning, statistics, econometrics, and thrash metal.

Results 3 issues of Ben Snyder

module.register_forward_hook only captures some layers when using torch.compile

5 comment

### 🐛 Describe the bug I've written a simple hook to save activations on the forward pass ``` def save_activations(name, mod, inp, out): activations[name] = out forward_handles = {} for...

triaged
oncall: pt2

added GPU environment support

1 comment

*Issue #, if available:* Issue https://github.com/aws-samples/sagemaker-studio-image-build-cli/issues/10 *Description of changes:* Adds `--environment` option to set `LINUX_GPU_CONTAINER` when build involves compiling Cuda operations. By submitting this pull request, I confirm that you...

Add support for GPU build environment

Currently build environment is hardcoded to `LINUX_CONTAINER`. This will cause builds that involve compiling Cuda operations to fail (for example, Docker images that include custom PyTorch or Tensorflow operations). Adding...

enhancement