Systems Research Engineer, GPU Programming
Develops and optimizes GPU-accelerated kernels and algorithms for ML/AI applications, co-designing with modeling, hardware, and software teams. Requires strong GPU programming expertise in CUDA/Triton and knowledge of ML models.
About the job
Responsibilities
- Optimize and fine-tune GPU code to achieve better performance and scalability
- Collaborate with cross-functional teams to integrate GPU-accelerated solutions into existing software systems
- Stay up-to-date with the latest advancements in GPU programming techniques and technologies
Requirements
- Strong background in GPU programming and parallel computing, such as CUDA and/or Triton
- Knowledge of ML/AI applications and models
- Knowledge of performance profiling and optimization tools for GPU programming
- Excellent problem-solving and analytical skills
- Bachelor's, Master's, or Ph.D. degree in Computer Science, Electrical Engineering, or equivalent practical experiences
Compensation
US base salary range: $160,000 - $230,000 + equity + benefits. Competitive compensation with flexibility in remote work.
Skills
CUDA, Triton, Gpu Programming, Parallel Computing, Ml/Ai, Performance Profiling, Optimization Tools
Similar jobs
AI Research jobs
Counsel Health
New York, NYResearch Scientist developing and evaluating health-focused AI models, large language models, and agentic systems for clinical applications. The role requires advanced research experience, strong coding skills, healthcare or clinical-data experience, and top-tier AI/ML publications.
AI Digest
Remote
Conduct research on long-horizon, multi-agent AI behavior by designing agent environments, analyzing large-scale data, and running experiments. The role requires strong research judgment, rapid execution, independence, and familiarity with current AI developments.
AI Digest
Remote
Build, optimize, and evaluate long-running and multi-agent AI systems, along with tools for monitoring and analyzing their real-world behavior. The role requires software engineering experience with coding agents, strong independence, and familiarity with current AI developments.
Hightouch
United States
Develops experimental AI techniques and prototypes for agentic marketing applications, with emphasis on image and video generation. The role requires strong backend or probabilistic systems expertise, quantitative thinking, creativity with LLM applications, and product intuition.
Greptile
San Francisco, CA
The Research Engineer will apply advances in agents and language models to build and evaluate multi-agent systems for automated code validation and review. The role requires a computer science or equivalent background, research experience, strong programming skills, and product intuition.