hey hey hey the world would be a better place if we all used JAX instead :)