Research Engineer, AI/ML Systems

Lightning AI builds an end-to-end platform for developing, training, and deploying AI systems, including tools for experimentation, large-scale compute, and production inference. The Research Engineer will focus on post-training models and the systems supporting them, while developing software, tooling, infrastructure, and workflows that improve AI research, development, evaluation, and deployment.

Responsibilities:

• Develop and post-train models, while building and improving the systems and workflows needed to run, evaluate, debug, and scale training workloads
• Build software, tooling, and platform capabilities that improve how researchers, developers, and customers develop, train, and deploy AI systems
• Contribute to Lightning’s open-source projects by building new features, improving existing functionality, and collaborating with the broader developer community
• Work across deep learning systems, developer tooling, backend services, and platform infrastructure to solve a wide variety of engineering challenges
• Collaborate directly with customers to understand real-world AI workloads, investigate technical challenges, and translate those learnings into reusable product and platform improvements
• Prototype new ideas, evaluate approaches, and turn successful experiments into production-quality software
• Partner closely with research, product, and infrastructure engineering teams to improve developer experience, AI workflows, and platform capabilities
• Debug complex technical problems spanning machine learning, distributed systems, backend software, and developer tooling
• Learn new technologies quickly and contribute wherever your skills can have the greatest impact as team priorities evolve

Requirements:

• Experience building, training, evaluating, or experimenting with deep learning models
• Hands-on experience with deep learning frameworks such as PyTorch
• Strong software engineering fundamentals building software and debugging and problem-solving skills, with the ability to investigate unfamiliar technical challenges
• Curiosity, initiative, and a demonstrated ability to quickly learn new technologies and technical domains
• Excellent communication and collaboration skills, including the ability to work effectively across research, product, infrastructure, and customer-facing engagements
• Comfortable working in fast-moving, ambiguous environments where priorities evolve over time
• Bachelor's degree in Computer Science, Engineering, or a related technical field, or equivalent practical experience
• Experience with model training at scale, including distributed training, performance optimization, training stability, and/or large-scale experimentation
• Experience with transformer-based language models or modern generative AI systems
• Experience with distributed systems, cloud infrastructure, or large-scale machine learning workloads
• Familiarity with technologies such as CUDA, Hugging Face, DeepSpeed, FSDP, Triton, vLLM, SGLang, NVIDIA Molt, or related AI infrastructure tooling
• Experience contributing to open-source software or conducting research through academia, industry, or meaningful independent projects
• Startup experience or experience working on highly cross-functional engineering teams
• Master's degree or higher in Computer Science, Machine Learning, AI, or a related field