pytree TODO benchmark vs pytree of jax, pytorch, tensorflow C++ tests build for Win, Mac for publishing