Machine Learning Engineer

The role

You will build software to support the machine learning development cycle from data generation to deployment and monitoring. You will also be involved in research efforts including publishing research and contributing ideas.

What you will do

• Build robust data pipelines that mine the Internet at scale and generate millions of synthetic text examples
• Manage distributed infrastructure for multi-GPU LLM training
• Profiling and optimizing training and inference code
• Deploy efficient inference pipelines for serving LLMs at scale

Required qualifications

• B.S./M.S. in Computer Science or related areas
• Strong programming skills in Python and modern ML frameworks
• Excellent understanding of transformers and LLM fundamentals
• Comfort working across research and engineering boundaries

Nice to have

• Experience with NVIDIA GPU programming and CUDA
• Experience with distributed training frameworks, such as DeepSpeed, FSDL, Ray
• Experience with inference frameworks like vLLM
• Experience with large-scale data processing (Spark, Beam) and orchestration (Airflow)
• Experience with MLOps and experiment tracking
• Experience with DevOps tools
• Familiarity with cloud-based infrastructure (AWS/GCP)

Before you apply

• Please don't use AI to answer any of the questions as part of your application

About the company

Pangram Labs builds AI detection systems to protect authenticity and preserve human connection. They publish research on state-of-the-art AI detection techniques and algorithms.