Add an API to implicitly compile modules?

[pranavm-nvidia]

We should consider adding an API to implicitly compile modules the first time they are called. This makes it so the user does not need to manually call tp.compile and should make interop with other frameworks a bit easier.

The API could optionally allow users to specify a shape range to support dynamic shapes.

[pranavm-nvidia]

Additional context for the ask:

It would 100% be a convenience API - it'd be interesting in combination with the auto dynamic shape option described above. And then again if I only port some of the submodules to tripy that are performance critical, I don't want to go ahead and compile those individually - especially as I would first need to figure out the intermediate shapes. Rather having the option to add a decorator or member variable (@auto_compile module.auto_compile=True ) and then triggering the compile during first forward call.