Research Scientist (Test Time Compute) - Remote

Overview

We are seeking an exceptional AI Research Scientist to join Naptha AI at the ground floor, focusing on advancing the state of the art in test time compute optimization for large language models. In this role, you will be responsible for researching and developing novel approaches to improve inference efficiency, reduce computational requirements, and enhance model performance at deployment.

In Short

• Design and implement novel architectures for efficient model inference
• Develop frameworks for model compression and quantization
• Research approaches to optimize test-time computation across different hardware
• Create efficient protocols for distributed inference and resource management
• Implement and test new ideas through rapid prototyping

Requirements

• Strong background in machine learning and systems optimization
• Deep understanding of model compression and efficient inference techniques
• Hands-on experience with modern ML frameworks and deployment tools
• Experience with ML infrastructure and hardware acceleration
• Track record of implementing efficient ML systems
• Excellent programming skills (Python required, C++/CUDA a plus)
• Strong analytical and problem-solving abilities
• PhD in Machine Learning, Computer Science, Mathematics, or equivalent experience is a plus
• Published research in relevant fields is a plus

Benefits

• Competitive salary with significant equity stake
• Remote-first work environment
• Full medical, dental, and vision coverage
• Flexible PTO policy
• Learning and development budget
• Conference and research publication support
• Home office setup allowance