Commits · c44d2d9efa34901b84ec581a8f078e4a4e2d3581 · OpenDAS / torch-harmonics

21 Jul, 2025 34 commits
- updated docstring · c44d2d9e
  Andrea Paris authored Jul 17, 2025
  
  c44d2d9e
- updated docstring · 43e3e720
  Andrea Paris authored Jul 17, 2025
  
  43e3e720
- simplified docstrings for test classes · c44d4b23
  Andrea Paris authored Jul 17, 2025
  
  c44d4b23
- cleanup of unnecessary docstrings · ca46b9d2
  Andrea Paris authored Jul 16, 2025
  
  ca46b9d2
- removed docstrings from internal functions · 9c26a6d8
  Andrea Paris authored Jul 16, 2025
  
  9c26a6d8
- removed docstrings from backward passes · 328200ab
  Andrea Paris authored Jul 16, 2025
  
  328200ab
- removed docstrings from _init_weights · d70dee87
  Andrea Paris authored Jul 16, 2025
  
  d70dee87
- removed docstrings from examples · b17bfdc4
  Andrea Paris authored Jul 16, 2025
  
  b17bfdc4
- removed docstrings from forward passes · 901e8635
  Andrea Paris authored Jul 16, 2025
  
  901e8635
- removed docstrings from autograd functions · 6373534a
  Andrea Paris authored Jul 16, 2025
  
  6373534a
- update documentation · 95fc83a0
  root authored Jul 16, 2025
  
  95fc83a0
- cleanup of merge conflict resolution · be8c6719
  Andrea Paris authored Jul 16, 2025
  
  be8c6719
- Improved docstrings in tests · 82e7860d
  apaaris authored Jun 30, 2025
  
  82e7860d
- Improved docstrings in examples · 290da8e0
  apaaris authored Jun 30, 2025
  
  290da8e0
- Improved docstrings in torch_harmonics/examples · 6f3250cb
  apaaris authored Jun 30, 2025
  
  6f3250cb
- Improved docstrings in distributed · 50ebe96f
  apaaris authored Jun 30, 2025
  
  50ebe96f
- Improved docstrings in sht · 10cf65f6
  apaaris authored Jun 30, 2025
  
  10cf65f6
- Improved docstrings in quadrature · 742499ae
  apaaris authored Jun 30, 2025
  
  742499ae
- Improved docstrings in legendre · c4995635
  apaaris authored Jun 30, 2025
  
  c4995635
- Improved docstrings in filter basis · 1c2859f2
  apaaris authored Jun 30, 2025
  
  1c2859f2
- Improved docstrings in convolution · eeda67aa
  apaaris authored Jun 30, 2025
  
  eeda67aa
- Improved docstrings in neighborhood attention · ffee67f9
  apaaris authored Jun 30, 2025
  
  ffee67f9
- Improved docstrings in disco convolution · 913e80d4
  apaaris authored Jun 30, 2025
  
  913e80d4
- Corrected docstrings in _layers.py · 313b1b73
  apaaris authored Jun 26, 2025
  
  313b1b73
- Added docstrings to many methods · e4879676
  apaaris authored Jun 26, 2025
  
  e4879676
- Merge pull request #93 from NVIDIA/tkurth/device-fixes · b5c410c0
  Thorsten Kurth authored Jul 21, 2025
```
Tkurth/device fixes
```
  b5c410c0
- renamings · 3d604f85
  Thorsten Kurth authored Jul 21, 2025
  
  3d604f85
- replacing more torch.tensor with torch.as_tensor · 0767c39c
  Thorsten Kurth authored Jul 21, 2025
  
  0767c39c
- and one more · 5a6d52dd
  Thorsten Kurth authored Jul 21, 2025
  
  5a6d52dd
- fixing some more missing device statements · 30f7802b
  Thorsten Kurth authored Jul 21, 2025
  
  30f7802b
- adding more missing device statements · f30ec30a
  Thorsten Kurth authored Jul 21, 2025
  
  f30ec30a
- adding device args to more functions · ea8d1a2e
  Thorsten Kurth authored Jul 21, 2025
  
  ea8d1a2e
- making disco helpers GPU ready · c877cda6
  Thorsten Kurth authored Jul 21, 2025
  
  c877cda6
- small device fix · 00064117
  Thorsten Kurth authored Jul 21, 2025
  
  00064117
17 Jul, 2025 1 commit
- Merge pull request #91 from NVIDIA/maurob/devel · 4aaff021
  Thorsten Kurth authored Jul 17, 2025
```
Attention Backward improvement
```
  4aaff021
16 Jul, 2025 5 commits
- Renamed the template parameter to a simpler name (it's the number of warps per... · fa58767d
  Mauro Bisson authored Jul 16, 2025
```
Renamed the template parameter to a simpler name (it's the number of warps per tile used in the permutation).
```
  fa58767d
- small bf16 fix for w11 loss · 763d4371
  Andrea Paris authored Jul 16, 2025
  
  763d4371
- removing commented code · 6ac50e26
  Thorsten Kurth authored Jul 16, 2025
  
  6ac50e26
- cleanup with contiguous checks · 45fc2a46
  Thorsten Kurth authored Jul 16, 2025
  
  45fc2a46
- Removed stale comments. · 51200bda
  Mauro Bisson authored Jul 11, 2025
  
  51200bda