Machine Learning Research Engineer

About Voltai:Voltai is an AI company building frontier foundation models and agentic systems for semiconductor and electronics design.We are one of the fastest-growing startups (in terms of valuation and revenue bookings) and leader in AI for electronics design. Backed by the world's top VC firm, we are working with some of the world’s largest semiconductor and electronics companies.The founding team consists of Olympiad medalists, ICPC world champions, and university professors. In addition, our business team has a proven track record, having scaled revenue from $0 to $400M in just four years at their previous company.
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: 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.