100% remote role Need to work as per EST No agency or C2C will NOT be considered and Visa sponsorship is not available nor provided Machine Learning Engineer: Framework Migration &Systems Optimization (PyTorch to JAX) We are seeking a specialized Machine Learning Engineer with deep expertise in the high-performance AI stack. This role isn''t just about "translating" code; it s about re-architecting Large Language Models (LLMs) to thrive in a JAX-native environment, specifically targeting TPU an