Senior Machine Learning Engineer - Remote

Overview

This role involves working with cutting-edge AI cloud infrastructure to enhance methodologies for training and deploying large language models.

In Short

• Enhance fine-tuning methodologies for LLMs.
• Research inference optimization techniques.
• Re-implement state-of-the-art LLM architectures in JAX.
• Deep understanding of machine learning and reinforcement learning.
• Expertise in deep learning for language processing.
• Experience training large models on multiple computational nodes.
• Strong software engineering skills, primarily in Python.
• Experience with modern deep learning frameworks, especially JAX.
• Proficiency in software engineering approaches like CI/CD.
• Strong communication and leadership skills.

Requirements

• Experience with language models or similar NLP technologies is a plus.
• Familiarity with concepts in LLM space.
• Experience in building and delivering products in dynamic environments.
• Strong engineering skills for large distributed systems.
• Open-source projects that demonstrate engineering skills.
• Excellent command of English and superior communication skills.

Benefits

• Competitive salary and comprehensive benefits.
• Opportunities for professional growth.
• Hybrid working arrangements.
• A dynamic and collaborative work environment.