Principal AI / ML Engineer

Principal Machine Learning Engineer

Our client is a digital consumer product. They're building social features using LLMs and Machine Learning to add to user experience. We're looking for a Principal Machine Learning Engineer to build AI features for the family.

Qualifications

• A confident craftsperson who possesses problem-solving tools and can discuss multiple approaches while preferring the best approach given the constraints.
• A scientific artist and artistic scientist who can navigate complexity and bring clarity to intricate problems while appreciating simplicity as the highest sophistication.
• A pragmatic optimizer who identifies the need for change, articulates it, and drives incremental improvements toward ambitious goals.
• An advocate for team collaboration and open communication.
• Possesses comprehensive knowledge of machine learning frameworks (e.g., TensorFlow, PyTorch) and advanced NLP techniques.
• Significant experience in AI/ML engineering with a strong data science background.
• Demonstrated ability in designing and training LLMs.
• Expertise in Databricks/Apache Spark for large-scale data processing in AI/ML applications.
• Proficiency with AWS Bedrock and Comprehend.
• Demonstrated leadership in AI/ML solutions in a commercial environment.
• Proficiency in programming languages such as R, Python, and Java, with a focus on data exploration and processing.
• Proven ability to lead and inspire teams of senior engineers and data scientists.
• Effective individual contributor within a collaborative team environment.
• Strong analytical, problem-solving, and communication skills.

Responsibilities

• Collaborate with the Data team to transform Data Lakehouse potential into actionable insights and services that enhance user experiences.
• Shape AI/ML strategies, particularly in LLMs, to drive business outcomes and establish data science principles.
• Implement technologies in AWS and Databricks for handling and processing large datasets for advanced language model training.
• Lead data science practices, including model selection and training methodology.
• Build, train, manage, and host LLMs using Data Lakehouse data.
• Optimize AI system performance through testing and tuning, leveraging data science and Databricks methodologies.
• Ensure ethical standards and data privacy compliance in AI solutions.
• Provide specialized AI/ML expertise to guide the development and optimization of AI features.
• Stay updated on the latest AI/ML and data science trends, especially in LLMs and AWS, Databricks/Apache Spark technology.

Benefits

• Full Medical Coverage: OFW covers 100% of premiums for employees and their families.
• 401k: Up to a 4% match with immediate vesting.
• 12 weeks of paid leave for all new parents.
• Learning & Development stipend for employees.
• Supportive and flexible working environment, allowing remote work from anywhere.