Hiring: Research Engineer, AI/ML Systems Company: Lightning AI Location: San Francisco, CA Job Posted Time: 2026-09-10 08:36:29 Employment Type: Remote Target Skills & Keywords : Deep Learning, Hugging Face, Machine Learning, PyTorch, Triton, vLLM About the job Experience: •Operational familiarity with technologies such as CUDA, Hugging Face, DeepSpeed, FSDP, Triton, vLLM, SGLang, NVIDIA Molt, or related AI infrastructure tooling. •Startup experience or experience working on highly cross-functional engineering teams. Required Skills: •Who enjoys solving difficult technical problems and building across the AI stack to join our Research Engineering function here at Lightning. •This role is hybrid with a minimum of 2 in-office days per week in San Francisco, Seattle, NYC, or London, with fully remote work considered for candidates outside of our office hub locations. All employees participate in occasional team and company offsites. •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. Qualifications: •Applied hands-on capability in 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. •Master's degree or higher in Computer Science, Machine Learning, AI, or a related field. Compensation: •$165,000 - $310,000 / year •Competitive benefits and rewards package Interested candidates, please apply directly through the job posting on company's career page or try via AI auto apply on this platform. Don't miss this opportunity to join a forward-thinking team!