Machine Learning Engineer

Pangram Labs builds AI detection systems and develops research and products focused on protecting authenticity. The Machine Learning Engineer will build software supporting the full machine learning development cycle, including data generation, model training, deployment, and production monitoring, while contributing to research and innovation.

What you'll do:

• 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

What they're looking for:

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