pytorch/torch/nn
Basil Wong aafc4b6188 Do not depend on numpy during the import (#150816)
Summary:
Related issue: https://github.com/pytorch/pytorch/issues/149681

We can follow up with a different implementation that does not use numpy(potentially with Torch primitives).

Test Plan:
pending:

contbuild & OSS CI

Differential Revision: D72609835

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150816
Approved by: https://github.com/jerryzh168, https://github.com/cyyever, https://github.com/albanD
2025-04-08 18:12:53 +00:00
..
attention [FlexAttention] Remove dead code (#150575) 2025-04-03 01:46:19 +00:00
backends
intrinsic
modules Add plot for torch.nn.Threshold and torch.nn.GLU (#150171) 2025-04-08 03:55:37 +00:00
parallel [BE][Ez]: Use itertools.chain.from_iterable when possible (#148190) 2025-03-06 20:37:06 +00:00
qat
quantizable
quantized
utils Do not depend on numpy during the import (#150816) 2025-04-08 18:12:53 +00:00
__init__.py
_reduction.py
common_types.py [5/N] Apply Ruff fixes and pyupgrade to Python 3.9 (#144205) 2025-01-15 04:00:47 +00:00
cpp.py
functional.py Fix torch.nn.functional.one_hot param num_classes optional description (#146470) 2025-02-06 07:48:05 +00:00
functional.pyi.in Use Python 3.9 typing (#148157) 2025-03-04 03:09:55 +00:00
grad.py
init.py
parameter.py remove typo in UninitializedParameter docstring (#140197) 2024-11-15 23:26:23 +00:00
parameter.pyi remove allow-untyped-defs from torch/nn/parameter.pyi (#144654) 2025-01-13 19:02:31 +00:00