[README]

Author: Kye

README.md

@@ -13,6 +13,31 @@ pip install mambabyte
 
 # Usage
 ```python
+import torch 
+from mambabyte import MambaConfig, Mamba
+
+x = torch.randn(2, 3, 4)
+config = MambaConfig(
+    dim = 4,
+    depth = 3,
+    dt_rank = 2,
+    d_state = 2,
+    expand_factor = 2,
+    d_conv = 3,
+    dt_min = 0.001,
+    dt_max = 0.1,
+    dt_init = "random",
+    dt_scale = 1.0,
+    bias = False,
+    conv_bias = True,
+    pscan = True
+)
+
+model = Mamba(config)
+
+out = model(x)
+
+print(out)
 
 ```
 

example.py

@@ -0,0 +1,26 @@
+import torch 
+from mambabyte import MambaConfig, Mamba
+
+x = torch.randn(2, 3, 4)
+
+config = MambaConfig(
+    dim = 4,
+    depth = 3,
+    dt_rank = 2,
+    d_state = 2,
+    expand_factor = 2,
+    d_conv = 3,
+    dt_min = 0.001,
+    dt_max = 0.1,
+    dt_init = "random",
+    dt_scale = 1.0,
+    bias = False,
+    conv_bias = True,
+    pscan = True
+)
+
+model = Mamba(config)
+
+out = model(x)
+
+print(out)