Description:

Join a cutting-edge AI research initiative focused on Generative AI and Large Language Models.
Collaborate with AI researchers and engineering teams to enhance model reasoning and MLOps infrastructure.

Requirements:

Minimum 2 years of experience in MLOps, Machine Learning Infrastructure, or ML Systems Engineering.
Hands-on experience with JAX and/or PyTorch in large-scale environments.
Practical experience with custom GPU kernels using Pallas or Triton.
Strong understanding of distributed training systems and scalable ML infrastructure.
Excellent written communication skills.

Skills