{"schemaVersion":"jobsearcher.job.v1","id":"c0db75412a0330e70b653f96","url":"https://jobsearcher.com/jobs/c0db75412a0330e70b653f96","canonicalUrl":"https://jobsearcher.com/jobs/c0db75412a0330e70b653f96","title":"MLOps Engineer (JAX, PyTorch, Pallas/Triton) | Remote","description":"Work SnapshotType: W2Location: United StatesCommitment: ~30+ hours/weekWhat You Ll Be DoingGuide engineering and research teams on MLOps, ML infrastructure, and training system optimizationDesign challenging tasks and develop structured solutions for ML systems and infrastructure workflowsEvaluate MLOps tasks and provide detailed technical feedbackDevelop evaluation frameworks for training pipelines, distributed systems, and kernel-level optimizationWork on large-scale AI model infrastructure using modern ML frameworksCollaborate with subject matter experts to ensure accuracy and consistency in training dataWhat We Re Looking ForStrong experience in ML infrastructure, MLOps, or ML systems engineeringStrong experience with JAX and/or PyTorch in production environmentsStrong experience in writing or optimizing GPU kernels using Pallas or TritonStrong understanding of distributed systems and training pipeline designStrong analytical and problem-solving skills in ML systems optimizationAbility to communicate complex technical decisions clearly in written formAbility to work consistently during weekdays in a remote environmentHow To ApplyUpload resumeInterviewSubmit form","company":"Codegeniusrecruit","rawCompany":"codegeniusrecruit","isRemote":true,"isActive":false,"createdAt":"2026-05-11T20:23:23.734Z","occupations":[{"code":"15-1252.00","title":"Software Developers","slug":"software-developers"},{"code":"15-1299.08","title":"Computer Systems Engineers/Architects","slug":"computer-systems-engineers-architects"},{"code":"15-1221.00","title":"Computer and Information Research Scientists","slug":"computer-and-information-research-scientists"}],"industries":[{"code":"541512","title":"Computer Systems Design Services","slug":"computer-systems-design-services"},{"code":"541511","title":"Custom Computer Programming Services","slug":"custom-computer-programming-services"},{"code":"513210","title":"Software Publishers","slug":"software-publishers"}],"jobPosting":{"@context":"https://schema.org","@type":"JobPosting","title":"MLOps Engineer (JAX, PyTorch, Pallas/Triton) | Remote","description":"Work SnapshotType: W2Location: United StatesCommitment: ~30+ hours/weekWhat You Ll Be DoingGuide engineering and research teams on MLOps, ML infrastructure, and training system optimizationDesign challenging tasks and develop structured solutions for ML systems and infrastructure workflowsEvaluate MLOps tasks and provide detailed technical feedbackDevelop evaluation frameworks for training pipelines, distributed systems, and kernel-level optimizationWork on large-scale AI model infrastructure using modern ML frameworksCollaborate with subject matter experts to ensure accuracy and consistency in training dataWhat We Re Looking ForStrong experience in ML infrastructure, MLOps, or ML systems engineeringStrong experience with JAX and/or PyTorch in production environmentsStrong experience in writing or optimizing GPU kernels using Pallas or TritonStrong understanding of distributed systems and training pipeline designStrong analytical and problem-solving skills in ML systems optimizationAbility to communicate complex technical decisions clearly in written formAbility to work consistently during weekdays in a remote environmentHow To ApplyUpload resumeInterviewSubmit form","datePosted":"2026-05-11T20:23:23.734Z","dateModified":"2026-05-11T20:23:23.734Z","hiringOrganization":{"@type":"Organization","name":"Codegeniusrecruit","sameAs":"https://jobsearcher.com"},"jobLocationType":"TELECOMMUTE","applicantLocationRequirements":{"@type":"Country","name":"US"},"identifier":{"@type":"PropertyValue","name":"JobSearcher","value":"c0db75412a0330e70b653f96"},"url":"https://jobsearcher.com/jobs/c0db75412a0330e70b653f96"}}