Job Closed
This listing is no longer active.
Next-generation content creation with artificial intelligence
Member of Technical Staff, Research Engineer – GPU Performance
Location
United States
Posted
67 days ago
Salary
$270K - $370K / year
Seniority
Lead
Job Description
Member of Technical Staff, Research Engineer – GPU Performance
Runway
• Help world models train faster and run more efficiently. • Profile, optimize, and rearchitect systems that turn research ideas into models that run at scale and in real time. • Optimize training throughput across large GPU clusters. • Design and maintain distributed training infrastructure. • Profile and accelerate inference pipelines for real-time multimodal generation. • Optimize and scale training infrastructure to improve efficiency and reliability. • Contribute to the entire stack, from low-level kernel optimizations to high-level model design.
Job Requirements
- 4+ years of experience in systems engineering, ML infrastructure, or performance optimization for deep learning.
- Familiarity with GPU kernel development (CUDA, Triton, CUTLASS) and distributed systems (NCCL, collective communication, model parallelism).
- Experience with ML framework internals (PyTorch, JAX) and mixed-precision / low-precision techniques (FP8, INT8).
- Experience building and operating large-scale training infrastructure, including fault tolerance and cluster orchestration.
- Excitement about building AI that simulates the world — and making it performant enough to run in real time.
- Bonus if you have experience with torch’s compilation feature.
Benefits
- Salary range based on competitive market rates for our size, stage and industry.
- Pay equity for our team.
Related Guides
Related Categories
Related Job Pages
More Research Engineer Jobs
Machine Learning Research Engineer
The American Institute of Architects (AIA)Teaching you what architects & architecture do for you and your community. Representing more than 98k architects!
• Participate in developing cutting-edge machine learning and data analytic solutions in healthcare • Participate in the development, research, evaluation, and deployment of machine learning methodologies for real-time clinical decision systems, patient monitoring, and alerting solutions • Achieving long-term and ambitious product development goals while identifying intermediate milestones • Being independent problem solvers while operating in a collaborative environment
Research Engineer
Quokka.ioProactive mobile security that makes you, your customers, organization, and employees feel safe and secure.
• Research and develop novel solutions for identifying security and privacy issues in mobile apps and platforms. • Develop automated static/dynamic program analysis engines. • Research and systemize emerging trends in the mobile application security landscape.
Innovation Engineer
EnvisioHelping local governments and school districts to build trust with their communities.
• Deliver a steady cadence of meaningful proof-of-concept implementations exploring how emerging technologies can address specific product or operational challenges. • Evaluate emerging technologies in controlled environments to determine whether they are suitable for eventual integration into the core product. • Identify opportunities that leverage modern technologies (particularly AI) to amplify what we can already do, enabling faster delivery, improved efficiency, and new product possibilities. • Share prototypes, findings, and techniques with the engineering team to encourage experimentation and adoption of useful technologies, helping catalyze innovation across the organization. • Provide periodic insight to engineering leadership on emerging technologies and their implications for product capabilities, engineering workflows, and long-term strategy.
Staff ML Research Engineer
SandboxAQLeveraging AQ - the powerful compound effects of AI + Quantum technology
• Bring content of scientific papers into promising, scalable ML algorithms; and translate these into high-performing and robust scientific code • Lead the ideation, benchmarking, and execution of complex datasets and ML models, ensuring seamless integration into our large-scale simulation frameworks. • Implement advanced software and hardware optimizations to maximize the efficiency of ML pipelines across distributed cloud GPU environments. • Drive software through the entire product lifecycle—from foundational research and implementation to launch and long-term support—ensuring technical excellence at every stage.




