Joel Schlosser 869d629c0f Forward / backward NJT support for several activation functions (#140736) ...

Several activation functions were unimplemented due to missing `pointwise` tags. This PR adds them and corresponding backwards implementations.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/140736
Approved by: https://github.com/Skylion007, https://github.com/cpuhrsch
ghstack dependencies: #141500

2024-11-26 21:19:58 +00:00

..

_internal

Forward / backward NJT support for several activation functions (#140736)

2024-11-26 21:19:58 +00:00

__init__.py

Allow NJT by default for weights_only torch.load (take 2) (#140739)

2024-11-19 02:44:53 +00:00