Inquiry About RouteNet Compatibility on ARM64 Linux

hey Miquel Ferriol Galmés,
I am interested in using your RouteNet repository inside a Docker container on an ARM64 Linux system (nvidia drive AGX). However, I encountered an issue with TensorFlow: the version specified in requirements.txt (tensorflow==2.6.*) does not seem to be compatible with ARM64.

Instead, I can only install tensorflow-aarch64 (e.g., version 2.11.1).
My questions are:

Will the model still work if I use tensorflow-aarch64 instead of tensorflow==2.6.*?
Are there any known limitations or modifications I should be aware of?
I would appreciate your insights. Thank you in advance for your help!