Machine Learning Research Scientist

About Voltai: Voltai is a Sequoia-backed AI company building agentic systems and frontier foundation models for semiconductor and electronics design. We're the fastest-growing and highest valued company in this space, with over 70% of the top semiconductor market already working with us. Our world-class team includes Stanford professors, Olympiad medalists, and the ex-CTO of Synopsys; our business leadership has scaled revenue in their previous companies to over $1.5B.
Key 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.

Required Skillsets:

• 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.

Our 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: Your health matters. We offer top-tier medical and dental insurance to keep you and your loved ones covered.
• Commitment to Your Growth: At Voltai, we’re dedicated to your continuous learning and development. Whether it’s through challenging projects or opportunities for professional advancement, we invest in your journey to becoming a leader in your field.
• Visa Sponsorship: We support international talent and offer visa sponsorship to help you join our team, no matter where you are in the world.