Partner Engineer, PyTorch

Menlo Park, CA

machine learning

natural language

reinforcement learning

Share:

Meta is seeking a PyTorch Partner Engineer to join Meta’s AI Partner Engineering team. This highly technical team collaborates with strategic partners, machine learning leaders across the industry, and major cloud service providers to build and launch innovative AI product services and experiences. In this role, you will engage with some of the world's most influential companies and work closely with our engineering and AI research teams. You will have industry experience working with PyTorch across a diverse range of deep learning and machine learning applications, including but not limited to natural language processing, computer vision, reinforcement learning, and other AI domains. You will play a key role in taking AI models from research to production, ensuring they are optimized and scalable for real-world applications.

Collaborate with partners to integrate and optimize PyTorch models for various AI applications.

Work closely with PyTorch engineering and research teams to support the deployment of AI solutions.

Define and execute on a strategy to drive adoption of PyTorch in research and production.

Work closely with cloud providers (e.g. AWS, Azure, GCP) and Meta’s strategic partners on AI models and PyTorch integrations.

Be a subject matter technical expert, and have strategic influence on partners and internal teams at Meta

Provide thought leadership, share best practices on PyTorch through forums such as medium blogs, white-papers, reference architectures and public-speaking events

Develop reference architectures, samples and other materials to share with the broader Meta developer community

Work with key partners and AI developer community to optimize existing generative AI models for improved performance, scalability, and efficiency.

Leads technical strategy and roadmap for long-term strategic partners with high level of autonomy / Demonstrated experience to own and navigate long term technical relationship with partners

Has good understanding of Responsible AI, proactively identifies risk to Meta and partners and drives involvement from the necessary cross functional partners

Effectively plan, lead and execute on projects with minimal direction & oversight from leadership

Proactively communicate feedback, updates, status to direct team as well as relevant cross-functional teams

Create clear and concise documentation, including technical specifications, best practices guides, and presentations, to communicate complex AI concepts to both technical and non-technical stakeholders internally and externally.

Minimum Qualifications

Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience.

5+ years experience as software engineer, technical consultant or partner/sales engineer

5+ years of experience in one or more of the following areas: Deep Learning, LLMs, NLP, Speech, Conversational AI,AI-Infrastructure, Fine-tuning and optimizations of PyTorch models.

Software development experience in languages like Python, Java, Go, Rust, C/C++. (at least one)

Experience with at least one deep learning framework such as PyTorch and/or JAX

Experience building cloud solutions on any cloud. Experience with Open Source cloud stacks like Kubernetes, Kubeflow, Docker containers.

Experience presenting to technical and business audiences

Preferred Qualifications

BS, MS, or Ph.D. degree in Computer Science or related quantitative field

Solid understanding of at least one Deep Learning framework (PyTorch, Tensorflow or MXNet)

Experience building Deep Learning - Computer Vision or Natural Language Processing or Recommender Systems models using the frameworks.

Apply relevant AI and machine learning techniques to build production ready PyTorch solutions with key partners.

Experience working with and building mindshare convincingly with any audience.

Experience working with internal and external partners

Data science background and experience manipulating/transforming data, model selection, model training, model optimization and deployment at scale.

Experience deploying production-grade machine learning solutions on public cloud platforms (like AWS)

Experience of launching a product / service or application into market is a plus

Knowledge of Mobile/IoT for deploying ML models at the edge is a plus

Contributed to an Open Source project, submitted PRs for features/ fixed bugs and/or created sample applications in OSS or participated in Kaggle competitions

For those who live in or expect to work from California if hired for this position, please click here for additional information.

About Meta

Meta builds technologies that help people connect, find communities, and grow businesses. When Facebook launched in 2004, it changed the way people connect. Apps like Messenger, Instagram and WhatsApp further empowered billions around the world. Now, Meta is moving beyond 2D screens toward immersive experiences like augmented and virtual reality to help build the next evolution in social technology. People who choose to build their careers by building with us at Meta help shape a future that will take us beyond what digital connection makes possible today—beyond the constraints of screens, the limits of distance, and even the rules of physics.

$147,000/year to $203,000/year + bonus + equity + benefits

Individual compensation is determined by skills, qualifications, experience, and location. Compensation details listed in this posting reflect the base hourly rate, monthly rate, or annual salary only, and do not include bonus, equity or sales incentives, if applicable. In addition to base compensation, Meta offers benefits. Learn more about benefits at Meta.

Equal Employment Opportunity

Meta is proud to be an Equal Employment Opportunity employer. We do not discriminate based upon race, religion, color, national origin, sex (including pregnancy, childbirth, reproductive health decisions, or related medical conditions), sexual orientation, gender identity, gender expression, age, status as a protected veteran, status as an individual with a disability, genetic information, political views or activity, or other applicable legally protected characteristics. You may view our Equal Employment Opportunity notice here.

Meta is committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures. If you need assistance or an accommodation due to a disability, fill out the Accommodations request form.

Please mention that you found this job on MoAIJobs, this helps us grow, thanks!