diff --git a/example.py b/example.py
new file mode 100644
index 0000000..6d4e051
--- /dev/null
+++ b/example.py
@@ -0,0 +1,10 @@
+import torch
+import torch.nn as nn
+
+class MyModel(nn.Module):
+    def __init__(self):
+        super().__init__()
+        self.fc = nn.Linear(10, 2)
+
+    def forward(self, x):
+        return self.fc(x)