Job descriptionProven experience with JAX and NumPy in real-world ML projectsStrong understanding of TensorFlow, including internals and model structureExperience with migrating models between frameworks and ensuring paritySolid grasp of ML training workflows, loop mechanics, and evaluation metricsStrong Python programming skills and clean, modular code practicesFamiliarity with ML experiment tracking and reproducibility tools (e.g., MLFlow, Weights & Biases)