MACHINE LEARNING RESEARCH SCIENTIST

$150,000 - $400,000/Yr

Voltai - Palo Alto, CA

Full-time - Mid Level

Palo Alto, CA

About the position

Voltai's mission is to re-build the physical world through developing super-intelligence to accelerate the pace of hardware innovation. Our focus is to build frontier models that could understand one of the world's most complex technologies—semiconductors and electronics. We are looking for a Machine Learning Research Scientist who thinks crazy ideas are normal, doesn't take 'No' as an answer, and thrives on the bursts of dopamine that come with startup responsibilities. As a Machine Learning Research Scientist, your ultimate goal will be to conduct cutting-edge research and build AI systems that deeply understand and innovate within the semiconductor and electronics domains, serving our customers. In this role, you will be responsible for developing new models, experimenting with novel architectures, and advancing the state of the art in AI for the semiconductor and electronics industries. Your work will involve both theoretical research and practical implementation, including designing and optimizing models, creating advanced algorithms, and publishing groundbreaking research. You will conduct original research in machine learning and AI with a focus on models that integrate multiple modalities, develop, train, and fine-tune large language models (LLMs) and multi-modal models (MMMs), and explore and scale trillion-parameter neural networks, along with smaller, specialized models. Additionally, you will design and implement models capable of solving discrete and continuous constraint reasoning tasks and graph-related challenges, develop and evaluate interconnected systems comprising LLMs, NLP models, and retrieval algorithms, and publish research findings in top AI conferences and journals, contributing to the academic and industry community.

Responsibilities

• Conduct original research in machine learning and AI with a focus on models that integrate multiple modalities.
• Develop, train, and fine-tune large language models (LLMs) and multi-modal models (MMMs).
• Explore and scale trillion-parameter neural networks, along with smaller, specialized models.
• Design and implement models capable of solving discrete and continuous constraint reasoning tasks and graph-related challenges.
• Develop and evaluate interconnected systems comprising LLMs, NLP models, and retrieval algorithms.
• Publish research findings in top AI conferences and journals, contributing to the academic and industry community.
• Collaborate with cross-functional teams to translate research outcomes into practical applications.
• Innovate and create data synthetically and collect it from human interactions for diverse tasks.
• Lead efforts in optimizing both open-source and proprietary models, considering various constraints.

Requirements

• Strong background in machine learning, including sequence modeling, generative models, and model architecture.
• Expertise in Pytorch, Python, CUDA, and Triton.
• Proven research experience with a strong publication record in top AI conferences and journals.
• Deep understanding of pre-training and fine-tuning large multi-modal models.
• Experience implementing research papers into production code.
• Familiarity with the latest state-of-the-art techniques, including prompting and inference-time search methods.
• Experience in developing and managing large-scale machine learning systems.
• Proficiency in configuring and optimizing hardware and operating systems for maximum performance.
• Experience building distributed training systems for AI models in high-performance computing (HPC) clusters.

Benefits

• Unlimited PTO: We trust you to manage your time and know when you need a break. Recharge when you need it, no questions asked.
• Comprehensive Health Coverage: We offer top-tier medical and dental insurance to keep you and your loved ones covered.
• Commitment to Your Growth: We invest in your journey to becoming a leader in your field through challenging projects and opportunities for professional advancement.