2 of 2 Remote/Hybrid Permanent JAX Jobs

awareness of latency and cost tradeoffs MLOps competency — experiment tracking, model versioning, evaluation pipelines, ML CI Experience with Python and modern ML frameworks (PyTorch, JAX) Excellent verbal and written English communication skills Excellent analytical and problem-solving skills A desire to innovate and push current practice Desirable: Experience with vision ...

experience strongly preferred Proven end-to-end ML experience: data pipelines, model development, deployment, and monitoring Strong Python and ML engineering skills (PyTorch or JAX) PhD in ML, Statistics, Computer Science, or related field (or equivalent research output) Ability to communicate complex ideas clearly to both technical and non-technical ...