{"schemaVersion":"jobsearcher.job.v1","id":"8250309472a89a6f9c117a7b","url":"https://jobsearcher.com/jobs/8250309472a89a6f9c117a7b","canonicalUrl":"https://jobsearcher.com/jobs/8250309472a89a6f9c117a7b","title":"JAX Expert | Remote","description":"Work SnapshotType: W-2 ContractCommitment: 40 hours per week, weekdaysWhat You'll Be DoingGuide research and engineering teams to close knowledge gaps and improve model performance across MLOps, training infrastructure, and ML framework-level topicsDesign and write accurate, well-structured solutions to MLOps and ML systems problems, generating high-quality training data for frontier systemsEvaluate tasks and solutions, providing clear written technical feedback with a focus on quality and consistencyDevelop detailed rubrics and evaluation frameworks to assess training pipeline design, distributed systems reasoning, and kernel-level optimizationCollaborate with subject matter experts to ensure accuracy and consistency across training data contributionsWhat We're Looking ForStrong experience in ML infrastructure, MLOps, or ML systems engineering at a top-tier organizationStrong experience in JAX at scale in a production environmentStrong experience writing or optimizing custom GPU kernels using Pallas or TritonStrong experience designing and evaluating domain-relevant technical tasks with clear, structured written outputStrong experience with distributed systems reasoning and training pipeline architectureHow To ApplyUpload resume","company":"Codegeniusrecruit","rawCompany":"codegeniusrecruit","isRemote":true,"isActive":false,"createdAt":"2026-06-26T13:52:00.289Z","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":"13-1151.00","title":"Training and Development Specialists","slug":"training-and-development-specialists"}],"industries":[{"code":"541511","title":"Custom Computer Programming Services","slug":"custom-computer-programming-services"},{"code":"541512","title":"Computer Systems Design Services","slug":"computer-systems-design-services"},{"code":"513210","title":"Software Publishers","slug":"software-publishers"}],"jobPosting":{"@context":"https://schema.org","@type":"JobPosting","title":"JAX Expert | Remote","description":"Work SnapshotType: W-2 ContractCommitment: 40 hours per week, weekdaysWhat You'll Be DoingGuide research and engineering teams to close knowledge gaps and improve model performance across MLOps, training infrastructure, and ML framework-level topicsDesign and write accurate, well-structured solutions to MLOps and ML systems problems, generating high-quality training data for frontier systemsEvaluate tasks and solutions, providing clear written technical feedback with a focus on quality and consistencyDevelop detailed rubrics and evaluation frameworks to assess training pipeline design, distributed systems reasoning, and kernel-level optimizationCollaborate with subject matter experts to ensure accuracy and consistency across training data contributionsWhat We're Looking ForStrong experience in ML infrastructure, MLOps, or ML systems engineering at a top-tier organizationStrong experience in JAX at scale in a production environmentStrong experience writing or optimizing custom GPU kernels using Pallas or TritonStrong experience designing and evaluating domain-relevant technical tasks with clear, structured written outputStrong experience with distributed systems reasoning and training pipeline architectureHow To ApplyUpload resume","datePosted":"2026-06-26T13:52:00.289Z","dateModified":"2026-06-26T13:52:00.289Z","hiringOrganization":{"@type":"Organization","name":"Codegeniusrecruit","sameAs":"https://jobsearcher.com"},"jobLocationType":"TELECOMMUTE","applicantLocationRequirements":{"@type":"Country","name":"US"},"identifier":{"@type":"PropertyValue","name":"JobSearcher","value":"8250309472a89a6f9c117a7b"},"url":"https://jobsearcher.com/jobs/8250309472a89a6f9c117a7b"}}