initial commit in branch developing

examples/jax_playground.py

@@ -1,21 +1,5 @@
-import jax
 import jax.numpy as jnp
-import numpy as np
-from jax import random
-from jax import vmap, jit
-from functools import partial
 
-from examples.time_utils import using_cprofile
+EMPTY_NODE = jnp.full((1, 5), jnp.nan)
 
-
-@jit
-def func(x, y):
-    return x + y
-
-
-a, b, c = jnp.array([1]), jnp.array([2]), jnp.array([3])
-li = [a, b, c]
-
-cpu_li = jax.device_get(li)
-
-print(cpu_li)
+print(EMPTY_NODE)