add MLP layers

build/torch-universal/MLP/__init__.py

@@ -0,0 +1,6 @@
+from . import layers
+
+
+__all__ = [
+    "layers",
+]

build/torch-universal/MLP/layers.py

@@ -0,0 +1,9 @@
+import torch
+import torch_npu
+
+
+class MLP(torch.nn.Module):
+    def forward(self, hidden_state):
+        gate_up = torch.cat((self.gate_proj(hidden_state), self.up_proj(hidden_state)), dim=-1)
+        down_proj = self.down_proj(torch_npu.npu_swiglu(gate_up))
+        return down_proj