2 of 2 Permanent JAX Jobs in the South East

experience, with at least 2 years in an ML or AI-adjacent role Strong Python skills and familiarity with major ML frameworks (PyTorch or JAX); comfortable reading and modifying model code Hands-on experience building and operating ML pipelines in production: data pipelines, training orchestration, evaluation, and serving Experience with ...

Lead AI Engineer Early employee at a high-growth AI startup Ideal Experience Strong Python software engineering skills Expertise with PyTorch and/or JAX Experience deploying machine learning systems into production Deep knowledge of LLMs and Generative AI Hands-on experience with LoRA, QLoRA, DPO, RLHF, SFT, or related ...