2024-02-24 Tests jax: basic jit jax optimization (a930a76e) Features start trying jax optimizations replace np with jnp, get working base, now to optimize as needed init jax src and tests fix typos