Google Coral NPU: A Platform for Edge AI. Compile TensorFlow and JAX models down to the hardware