Add TorchModuleWrapper

[soumik12345]

Add the implementation of TorchModuleWarpper for wrapping both vanilla and lazy torch.nn.Modules as keras_core.layers.Layers.

Linked Issue: #604

Linked Notebook:

[soumik12345]

Hi @fchollet
I am encountering this import error while trying to import Variable.

[ianstenbit]

Thanks for the PR! Excited about this functionality

[fchollet]

Thanks for the PR!

[soumik12345]

Hi @fchollet
Here are 2 issues with the current code:

1. During model.call for lazy torch modules, an error is encountered, stating

TypeError: Exception encountered when calling LazyClassifier.call().

missing a required argument: 'inputs'

Is the compute_output_spec function used incorrectly?

2. During model.fit another error is encountered for lazy modules, stating

RuntimeError: Unable to automatically build the model

Here's a colab to reportduce the issues: https://colab.research.google.com/drive/1UO8uY86Ff-lNq5_7vKeLAb5sK99UFDWJ?usp=sharing

[fchollet]

Thanks. I'll take it over.