Latest AI JobsBrowse Jobs
bjakcareer logo

Staff Machine Learning Engineer

bjakcareerUnited StatesActAI Engineering
Apply on company site
Salary
Not listed
Job type
Full-time
Work mode
Remote
Experience
Senior
Posted
28 Sept 2026
Status
Open · checked 7d ago

About this role

This role suits an experienced engineer who wants to own the execution layer of artificial intelligence systems and build proactive applications in the United States. The Staff Machine Learning Engineer position at bjakcareer involves turning research and model capabilities into reliable, scalable production systems.

Role Overview

The Staff Machine Learning Engineer position at bjakcareer focuses on creating proactive applications for users worldwide with minimal prompting. Located in the United States, this role requires handling the complete lifecycle of machine learning systems, from data ingestion to model deployment and inference.

Key Responsibilities

Manage the complete end-to-end machine learning infrastructure covering data, training, evaluation, inference, and deployment.
Develop and improve training and fine-tuning pipelines for large models.
Create evaluation frameworks to track capability, safety, robustness, and real-world performance.
Build high-performance inference systems focused on optimizing memory, GPU usage, cost, latency, and reliability.
Construct data systems that handle both real-world and synthetic training data.
Set up dependable production infrastructure for model deployment, monitoring, and ongoing enhancement.
Collaborate alongside research and application engineering teams to translate model capabilities into product features.
Execute pragmatic technical decisions and iterate quickly based on performance metrics.

Requirements

Prior experience deploying and shipping production-grade machine learning systems rather than research prototypes.
Comprehensive knowledge of modern large-model training, fine-tuning, inference, and evaluation.
Solid fundamentals in software engineering and systems design.
Background in operating machine learning workloads at scale, specifically with GPU-based infrastructure.
Strong technical judgment and independence in solving ambiguous problems.
Focus on experimentation, measurement, and shipping code.
Commitment to correctness, reliability, and high production standards.

Skills and Tools

Python
PyTorch
JAX
GPU-based training and inference systems

About the Company

bjakcareer builds proactive applications designed for billions of users who are not accustomed to complex prompting, aiming to bring intelligence to everyday workflows, conversations, and tasks. The company operates as a small, hands-on team with high talent density where engineers maintain broad ownership, make fast decisions, and work independently.

How to Apply

Interested candidates should visit the official bjakcareer careers page to submit an application.

Skills and tech stack

Questions candidates ask

What location is the Staff Machine Learning Engineer role based in?

The position is located in the United States.

What technologies are used in the tech stack?

The tech stack includes Python, PyTorch, JAX, and GPU-based training and inference systems.

How many interview rounds can candidates expect?

Candidates will go through three to four interviews.

What is the primary focus of the engineering work?

The work focuses on building proactive applications, establishing reliable production infrastructure, and turning model capabilities into scalable production systems.

Ready to apply?

The application is on the official bjakcareer careers page.

Apply on company site