GrepJob
Pangram Labs

Machine Learning Engineer

Pangram Labs
Apply
3 months ago
Brooklyn, NY, USAEntry Level / Mid Level

Base Salary

$150k - $250k/yr

Responsibilities

  • Build robust data pipelines to generate synthetic text examples for training detection models.
  • Manage distributed infrastructure for multi-GPU LLM training.
  • Profile and optimize 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.

Tech Stack

Apache AirflowApache BeamApache SparkAWSGoogle Cloud PlatformPython

Categories

AI & MLData EngineeringDevOps