Modifiable script - #1

Open

kamatmihir2002 wants to merge 10 commits into

Jtrep10:mainfrom

kamatmihir2002:modifiable_script

Commits on Oct 9, 2025

Added gitignore
kamatmihir2002
committed
9da5422
Added img folder to save plots
kamatmihir2002
committed
e8d5e95

Commits on Oct 10, 2025

Modify original NTK script to handle 2d NTK calculation, as well as variable depth
kamatmihir2002
committed
fd3777c
Update .gitignore
kamatmihir2002
committed
168a172
Modified for dynamic adjustment of gpu/cpu modes
kamatmihir2002
committed
ca951b8
Added 3d plots -- X, Y are inputs and Z is NTK value
kamatmihir2002
committed
4108535
Plot 3d surface of XY inputs vs NTK
kamatmihir2002
committed
5524ae2
Made generic loss wrapper so that script can handle any loss function
kamatmihir2002
committed
391d632
Clean up the code
kamatmihir2002
committed
1ebdc0a
Alphabetical order for img
kamatmihir2002
committed
576e690