Systems Research Engineer, GPU Programming

Job Description

Together AI is seeking a Systems Research Engineer specializing in GPU Programming. The candidate will develop and optimize GPU-accelerated kernels and algorithms for ML/AI applications. He/She will co-design GPU kernels and model architecture to enhance the performance and efficiency of AI systems, and collaborate with hardware and software teams to co-design efficient GPU architectures and programming models. The candidate will also stay up-to-date with advancements in GPU programming techniques.Role involves:

• Optimizing and fine-tuning GPU code for performance and scalability.
• Integrating GPU-accelerated solutions into software systems.
• Staying updated on GPU programming advancements.

Requirements:

• Strong background in GPU programming and parallel computing (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 experience.

Together AI offers:

• Competitive compensation.
• Startup equity.
• Health insurance and other benefits.
• Flexibility in terms of remote work.