Machine Learning Engineer, Natural Language Generation (NLG)

Apple - Cupertino, CA

Full-time - Mid Level

Cupertino, CA

Computer and Electronic Product Manufacturing

About the position

As a Machine Learning Engineer specializing in Natural Language Generation (NLG) at Apple, you will play a pivotal role in enhancing user interaction through advanced text generation technologies. This position is part of the Input Experience NLP team within Software Engineering, where your contributions will impact billions of users. The team is dedicated to redefining how users interact with generative models for text generation, focusing on creating seamless input experiences across all Apple platforms. Your work will involve building and optimizing offline NLP models tailored for various hardware backends and user interfaces, ensuring that these innovations align with Apple's commitment to privacy by processing data on-device. In this role, you will collaborate with a talented group of machine learning and software engineers, working on cutting-edge technologies such as context-augmented text rewriting, safety-controlled text composition, and personalized smart interactions. The team has a rich history in this domain, owning the NLP and ML text input stack that powers features like autocorrection and predictive typing. Your expertise will help amplify these technologies, making them accessible not only on Apple devices but also to third-party applications through the Natural-Language framework. This is an exciting opportunity to leverage your machine learning skills to create user experiences that resonate with a global audience, contributing to Apple's legacy of innovation and excellence.

Responsibilities

• Develop and optimize machine learning models for natural language generation.
• Collaborate with cross-functional teams to integrate ML technologies into user interfaces.
• Implement and refine algorithms for text generation, ensuring high performance and accuracy.
• Conduct experiments to benchmark and improve model performance and efficiency.
• Work on the full stack of machine learning applied to natural language processing (NLP).
• Contribute to the development of the NLP and ML text input stack for Apple platforms.

Requirements

• Strong machine learning fundamentals.
• Strong programming and communication skills.
• Ph.D. in Computer Science, Electrical Engineering, Physics, Statistics, or related field; or a Bachelor/Master's degree with 2 years of industry experience.
• Experience with implementing optimizers and regularizations in machine learning models.
• Proficiency in programming with high-quality code across complex repositories.

Nice-to-haves

• Familiarity with common NLP algorithms and applications such as tokenization and language modeling.
• Experience with multi-modal modeling and presenting technical plans and results.
• Ability to process large datasets efficiently and establish automated training and evaluation pipelines.

Benefits

• Health insurance coverage.
• 401k retirement savings plan.
• Paid holidays and vacation time.
• Professional development opportunities.