change a lot a lot a lot!!!!!!!

examples/b.py

@@ -0,0 +1,19 @@
+from enum import Enum
+from jax import jit
+
+class NetworkType(Enum):
+    ANN = 0
+    SNN = 1
+    LSTM = 2
+
+
+
+
+@jit
+def func(d):
+    return d[0] + 1
+
+
+d = {0: 1, 1: NetworkType.ANN.value}
+
+print(func(d))