Research Engineer, Machine Learning (AI)

TL;DR

Research Engineer, ML (AI): Building and optimizing large-scale learning systems that power open-weight models, focusing on enhancing the shared training framework, data pipelines, and cluster tooling. Focus on integrating cutting-edge research with production by streamlining evaluation and exposing APIs, as well as designing and benchmarking ML algorithms.

Location: Hybrid, Palo Alto, USA

Company

hirify.global democratizes AI through high-performance, optimized, open-source, and cutting-edge models, products, and solutions designed to meet enterprise and personal needs.

What you will do

• Accelerate researchers by taking on the heavy parts of large-scale ML pipelines and building robust tools.
• Interface cutting-edge research with production: integrate checkpoints, streamline evaluation, and expose APIs.
• Conduct experiments on the latest deep-learning techniques (sparsified 70 B + runs, distributed training on thousands of GPUs).
• Design, implement and benchmark ML algorithms; write clear, efficient code in Python.
• Deliver prototypes that become production-grade components for Le Chat and our enterprise API.

Requirements

• Master’s or PhD in Computer Science (or equivalent proven track record).
• 4 + years working on large-scale ML codebases.
• Hands-on with PyTorch, JAX or TensorFlow; comfortable with distributed training (DeepSpeed / FSDP / SLURM / K8s).
• Experience in deep learning, NLP or LLMs; bonus for CUDA or data-pipeline chops.
• Strong software-design instincts: testing, code review, CI/CD.
• Self-starter, low-ego, collaborative.

Culture & Benefits

• Competitive salary and equity.
• Healthcare: Medical/Dental/Vision covered for you and your family.
• Pension: 401K (6% matching).
• PTO: 18 days.
• Transportation: Reimburse office parking charges, or $120/month for public transport.
• Meal stipend: $400 monthly allowance for meals.