refactor(autogram): Fix outdated list of incompatible modules (#450)

* Transformer and helper modules are now compatible with batched engine. It somehow worked even with those modules in the list because Transformer and helper modules do not have direct params and thus do not get hooked themselves anyway.

Author: ValerianRey

src/torchjd/autogram/_engine.py

@@ -19,11 +19,6 @@
     nn.LazyBatchNorm3d,
     nn.SyncBatchNorm,
     nn.RNNBase,
-    nn.Transformer,
-    nn.TransformerEncoder,
-    nn.TransformerDecoder,
-    nn.TransformerEncoderLayer,
-    nn.TransformerDecoderLayer,
 )
 
 _TRACK_RUNNING_STATS_MODULE_TYPES = (