Remote Principal AI Research Engineer

Summary

Join our core AI team to research, design, and implement advancements in Large Language Models (LLMs) to optimize for relevance, latency, and privacy. You will tackle complex challenges related to model optimization, fine-tuning, and tool development, impacting millions of users worldwide.

Requirements

• Experience in designing and developing machine learning models, especially in natural language processing
• Deep understanding of transformer architecture, LLMs, and embedding models
• Strong proficiency in Python, including experience with ML/NLP libraries (e.g., TensorFlow, PyTorch, Hugging Face Transformers)
• Experience with at least one low-level system language (c, c++, rust)
• Experience with CUDA and high-performance computing
• Self-directed, ambitious, and eager to learn new skills and technologies
• Master’s or PhD in Computer Science, Artificial Intelligence, Machine Learning, or a related field

Responsibilities

• Optimize and fine-tune LLMs or embedding models
• Develop and enhance tools and frameworks built around model orchestration
• Research, design, and implement state-of-the-art algorithms to optimize models at scale for latency and efficiency
• Investigate complex NLP problems driven by customer use cases
• Implement hardware-specific optimizations (e.g., GPUs, TPUs, CUDA)

Preferred Qualifications

• Publications or contributions to projects in AI/ML, NLP, or related areas
• Experience with large-scale distributed training and optimization techniques
• Contributions to open-source AI projects

Benefits

• Competitive salaries and equity grants
• Unlimited time off to promote a healthy work-life balance
• H/D/V coverage along with 401K, FSA, and Commuter Benefits
• Frequent team celebrations and recreation events
• Learning and development opportunities