I would like to know if someone has testes JAX in an embedded GPU?
JAX (JAX reference documentation — JAX documentation).
Jax accelerates training and inference in cloud GPUs and TPUs. I do not know if it is faster than TensorRT or if it works with embedded GPUs and TPUs.
Has anyone tested it?