Magenta RealTime 2 was released with native support to JAX/MLX. The JAX code allows the code to run on TPU/GPUs, while the MLX version allows the code and the exciting apps Google released to run locally on Mac devices.