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