Neural Tangent Kernel integration + typo fix (#505)

* NTK weighting + typo fixing
* black code formatter + .rst docs

---------

Co-authored-by: Dario Coscia <dariocos99@gmail.com>
This commit is contained in:
Giuseppe Alessio D'Inverno
2025-03-19 12:44:07 -04:00
committed by Nicola Demo
parent 2b09cb95cf
commit 716d43f146
4 changed files with 147 additions and 0 deletions

View File

@@ -0,0 +1,9 @@
NeuralTangentKernelWeighting
=============================
.. currentmodule:: pina.loss.ntk_weighting
.. automodule:: pina.loss.ntk_weighting
.. autoclass:: NeuralTangentKernelWeighting
:members:
:show-inheritance: