Machine Learning Engineer

About the Company

Pangram Labs’ mission is to protect authenticity by building the world’s best AI detection systems. We publish research on state-of-the-art AI detection techniques and algorithms, and build products to bring our technology into people's daily lives.

Responsibilities

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

Requirements

• 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

Preferred Qualifications

• 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)