Machine Learning Research Engineer

voltai-com; Edison (CA); On-site; USD 180,000 - 250,000; Full time

Benefits offered by this job

Unlimited PTO

Health coverage

Meals & snacks

Professional growth

Visa sponsorship

Job summary

Voltai is seeking an experienced machine learning engineer to train foundation models and multi-modal systems, driving production-ready implementations across hardware designs and semiconductor workflows.

You will lead end-to-end model development, from data structure to optimization on HPC clusters, with a focus on scalable inference and state-of-the-art prompting techniques.

Qualifications

• ,

Responsibilities

• Train foundation models from scratch that integrate multiple modalities.
• Train and fine-tune large language models (LLMs) and multi-modal models (MMMs).
• Design and implement models capable of solving discrete and continuous constraint reasoning tasks and graph-related challenges.
• 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.

Skills

Software engineering

Sequence modeling

Generative models

Model architecture

Pre-training & fine-tuning

Production code

Prompting & inference search

Distributed ML systems

Hardware & HPC optimization

Tools

PyTorch

Python

CUDA

Triton

Job description

About Voltai

Voltai is the leading AI company building agentic systems and frontier foundation models for semiconductor and electronics design. Backed by Sequoia Capital, we’re putting AI in the hands of hardware engineers in over 70% of the world’s largest semiconductor and electronics companies to have effortless control over their next-generation chip and board designs, powering the future of automotive, industrial automation, consumer electronics, IoT, and semiconductor manufacturing.

About the Team

Our founding team consists of IOI/IPhO olympiad medalists, Stanford professors, ex-CTO of Synopsys, and our business leadership has scaled revenue in their previous companies to over $1.5bn. At Voltai, we are combining the world’s best talent in the intersection of software and hardware.

Key Responsibilities

• Train foundation models from scratch that integrate multiple modalities.
• Train and fine-tune large language models (LLMs) and multi-modal models (MMMs).
• Design and implement models capable of solving discrete and continuous constraint reasoning tasks and graph-related challenges.
• 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.

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.

Our Benefits

• Unlimited PTO: Recharge when you need it, no questions asked.
• Comprehensive Health Coverage: Medical, dental, and vision insurance for you and your dependents.
• Free Meals and Snacks: Daily lunches, dinners, and snacks in the office.
• Professional Growth: We invest in your continuous learning and offer opportunities to expand your skills.
• Visa Sponsorship: We welcome global talent and provide visa sponsorship to support qualified candidates.