Adds tp.equal, improves Module prints, fixes nanoGPT

[pranavm-nvidia]

• Adds a tp.equal API and updates tests to use it where applicable

• Cleans up printing of modules, adds automatic conversion to Parameter when setting attributes

• Updates nanoGPT test to use standard regexs