Research Engineer (AI)

TL;DR

Research Engineer (AI): Designing, optimizing, and maintaining compute foundations for large-scale language model training with an accent on developing high-performance ML kernels and improving distributed compute stacks. Focus on collaborating with researchers to bridge algorithmic design with hardware efficiency.

Location: San Francisco, California

Compensation: $350,000 - $475,000 USD

Company

hirify.global empowers humanity through advancing collaborative general intelligence.

What you will do

• Design and implement custom ML kernels for core LLM operations optimized for modern GPU architectures.
• Collaborate with research teams to align kernel-level optimizations with model architecture.
• Develop and maintain a library of reusable kernels and performance benchmarks.
• Contribute to infrastructure stability and scalability, ensuring reproducibility and high utilization of compute resources.
• Document and share insights through internal talks and open-source contributions.

Requirements

• Bachelor’s degree or equivalent experience in relevant fields.
• Strong engineering skills and ability to debug complex codebases.
• Understanding of deep learning frameworks like PyTorch and JAX.
• Proficiency in CUDA, CuTe, Triton, or other GPU programming frameworks.
• Demonstrated ability to analyze and optimize compute-intensive workloads.