Machine Learning Research Engineer

Overview

About Voltai

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 – semiconductor and electronics.

About the Role

We’re looking for a Machine Learning Research Engineer 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 Engineer, your ultimate goal will be to build AI systems that deeply understand and innovate within the semiconductor and electronics domains, serving our customers 🙌.

You will be responsible for training and fine-tuning large-scale models, including trillion-parameter neural networks, and designing systems that integrate various modalities. Your work will involve creating and optimizing both open-source and proprietary models, developing advanced search and retrieval algorithms, and building AI systems that solve complex reasoning tasks.

Key Responsibilities:
• Train foundation models from scratch that integrate multiple modalities.
• Train and fine-tune large language models (LLMs) and multi-modal models (MMMs).
• Develop 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.
• Structure and optimize data for model training, ensuring high performance and efficiency.
• Create data synthetically and collect it from human interactions for diverse tasks.
• Build and optimize both open-source and proprietary models, considering various constraints.
• Design and execute innovative search and retrieval algorithms.
• Work on building foundation models that drive various products within the electronics and semiconductor sectors.

Required Skillsets:
• Strong software engineering skills, particularly in Pytorch, Python, CUDA, and Triton.
• Deep expertise in machine learning, including sequence modeling, generative models, and model architecture.
• Experience in pre-training and fine-tuning large multi-modal models.
• Proven track record of publications in top AI conferences and competitions.
• 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.

Compensation Philosophy

At Voltai, we believe that exceptional work deserves exceptional rewards. Our compensation structure is designed to reflect the value each team member brings to our pioneering efforts in the semiconductor and AI industries. For this role, we anticipate the starting annual base salary to be within the range of $150,000 to $350,000, adjusted according to the candidate’s experience, expertise, and impact potential. While this range provides a general guideline, the final offer may vary to ensure we align our incentives with individual contributions and the long-term success of our mission.

Our Benefits

At Voltai, we believe in taking care of our team so they can focus on pushing the boundaries of innovation. Our benefits package is designed to support your well-being and fuel your professional growth.
• 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.

Company:

Voltai

Qualifications:

Language requirements:

Specific requirements:

Educational level:

Level of experience (years):

Senior (5+ years of experience)

Tagged as: Clustering, Industry, Language Modeling, Neural Networks, NLP, United States