Bjak logo

Staff Machine Learning Engineer

Bjak
Posted a month ago

LOCATION

United States · Remote

EXPERIENCE

Not specified

TYPE

FullTime

SKILLS REQUIRED

Machine LearningSystem DesignPytorchModel DevelopmentContinuous DeploymentData Pipelines

Job description

Overview

Own the execution layer of the company’s intelligence by turning research and model capabilities into reliable, scalable production ML systems. Work across data, training, evaluation, inference, and deployment in a hands-on leadership role.

What you'll do

  • Own end-to-end ML systems from data and training through evaluation, inference, and deployment.
  • Build and evolve training and fine-tuning pipelines for large models.
  • Design evaluations for capability, robustness, safety, and product performance.
  • Architect inference systems for latency, GPU utilization, memory, cost, and reliability.
  • Build pipelines for real-world and synthetic training data.
  • Establish infrastructure for deploying, monitoring, and improving models.
  • Partner with research and application engineering to improve products.

What you'll need

  • Experience building and shipping production ML systems.
  • Understanding of large-model training, fine-tuning, evaluation, and inference.
  • Strong software engineering and systems fundamentals.
  • Experience operating GPU-based ML workloads at meaningful scale.
  • Strong technical judgment and independent problem-solving ability.
  • Python and PyTorch or JAX experience.
  • Production-grade coding and attention to system correctness.

Details

  • Location: United States.
  • Interviews may be virtual and/or onsite.

Read the full description and apply on the company’s own careers page.

Stay safe

Hiring on Abekus is free for applicants

We never charge a fee, and employers are prohibited from doing so. If a recruiter asks for payment, please report them right away.

Staff Machine Learning Engineer