Remote Principal AI Research Engineer

Summary

Join our core AI team to research, design, and implement advancements in how companies run Large Language Models (LLMs) to optimize for relevance, latency, and privacy.

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
• 25 vacation days per year (in addition to bank holidays) to promote a healthy work-life balance
• Private health insurance
• Paid paternity leave
• Flexible working options
• Frequent team celebrations and recreation events
• Home internet & phone stipend
• Learning and development opportunities