This version includes the following update:
- Enhancing performance of dataset preprocessing by removing JIT compilation as well as using numpy instead of jax.numpy in edge vector function inside the neighbor analysis code (contributed by @bhcao in #42).
