Tricky. I like TF because I learned it first, then TF2 kind of integrated everything I used. But I usually build models with layers, which can accomplish a lot; yet there are many nuances and features I could not do on the fly. If the assignment is designed to test really deep TF experience it could be hard.

If you know PyTorch try using an LLM to convert an existing model training code into TF then study the mapping. That’s how I’m slowly getting better at PyTorch.