{"schemaVersion":"jobsearcher.job.v1","id":"4b8077ecedfa13676cc9c3af","url":"https://jobsearcher.com/jobs/4b8077ecedfa13676cc9c3af","canonicalUrl":"https://jobsearcher.com/jobs/4b8077ecedfa13676cc9c3af","title":"Machine Learning Infrastructure Engineer (Modeling)","description":"The ML Infrastructure team supports and accelerates PI’s core modeling efforts by building the systems that make large-scale training reliable, reproducible, and fast. The team works closely with research, data, and platform engineers to ensure models can scale from prototype to production-grade training runs\r\nOwn training/inference infrastructure: Design, implement, and maintain systems for large-scale model training, including scheduling, job management, checkpointing, and metrics/logging\r\nScale distributed training: Work with researchers to scale JAX-based training across TPU and GPU clusters with minimal friction\r\nOptimize performance: Profile and improve memory usage, device utilization, throughput, and distributed synchronization\r\nEnable rapid iteration: Build abstractions for launching, monitoring, debugging, and reproducing experiments\r\nPartner with researchers: Translate research needs into infra capabilities and guide best practices for training at scale\r\nContribute to core training code: Evolve JAX model and training code to support new architectures, modalities, and evaluation metrics#J-18808-Ljbffr","company":"Physical Intelligence","rawCompany":"physical intelligence","city":"Millbrae","state":"CA","isRemote":false,"isActive":false,"createdAt":"2026-09-23T01:13:52.214Z","occupations":[{"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"},{"code":"15-1252.00","title":"Software Developers","slug":"software-developers"}],"industries":[{"code":"518210","title":"Computing Infrastructure Providers, Data Processing, Web Hosting, and Related Services","slug":"computing-infrastructure-providers-data-processing-web-hosting-and-related-services"},{"code":"541512","title":"Computer Systems Design Services","slug":"computer-systems-design-services"},{"code":"541511","title":"Custom Computer Programming Services","slug":"custom-computer-programming-services"}],"jobPosting":{"@context":"https://schema.org","@type":"JobPosting","title":"Machine Learning Infrastructure Engineer (Modeling)","description":"The ML Infrastructure team supports and accelerates PI’s core modeling efforts by building the systems that make large-scale training reliable, reproducible, and fast. The team works closely with research, data, and platform engineers to ensure models can scale from prototype to production-grade training runs\r\nOwn training/inference infrastructure: Design, implement, and maintain systems for large-scale model training, including scheduling, job management, checkpointing, and metrics/logging\r\nScale distributed training: Work with researchers to scale JAX-based training across TPU and GPU clusters with minimal friction\r\nOptimize performance: Profile and improve memory usage, device utilization, throughput, and distributed synchronization\r\nEnable rapid iteration: Build abstractions for launching, monitoring, debugging, and reproducing experiments\r\nPartner with researchers: Translate research needs into infra capabilities and guide best practices for training at scale\r\nContribute to core training code: Evolve JAX model and training code to support new architectures, modalities, and evaluation metrics#J-18808-Ljbffr","datePosted":"2026-09-23T01:13:52.214Z","dateModified":"2026-09-23T01:13:52.214Z","hiringOrganization":{"@type":"Organization","name":"Physical Intelligence","sameAs":"https://jobsearcher.com"},"jobLocation":{"@type":"Place","address":{"@type":"PostalAddress","addressLocality":"Millbrae","addressRegion":"CA","addressCountry":"US"}},"identifier":{"@type":"PropertyValue","name":"JobSearcher","value":"4b8077ecedfa13676cc9c3af"},"url":"https://jobsearcher.com/jobs/4b8077ecedfa13676cc9c3af"}}