Senior Machine Learning Engineer

Bjak; New York (NY); On-site; USD 110,000 - 130,000; Full time

Job summary

A leading AI firm in New York is seeking an experienced machine learning engineer to build and manage impactful ML systems. This role requires ownership from data preparation to deployment, collaborating across functions, and mentoring other engineers. The ideal candidate has a profound understanding of modern ML models and writes production-quality code. The position offers a chance to be part of an innovative team focused on making AI impactful globally.

Qualifications

• Experience building ML systems used by real users.
• Strong understanding of ML model behavior in production.
• Ability to work independently and drive projects.

Responsibilities

• Build core ML systems for a proactive AI product.
• Own the entire work process from data preparation to deployment.
• Collaborate closely with cross-functional teams.

Skills

Built and shipped ML systems

Strong production-quality code

Ownership and independence

Fast learners

Clear communication

Tools

Python

PyTorch

JAX

Job description

About the Role

A1 is building a proactive AI system that carries work forward across conversations, tools, and time.

You will be a core owner of significant ML systems in production. This role is for engineers who can independently drive complex ML problems from idea to deployment and make strong trade-offs along the way.

Focus

• Build core ML systems that power a proactive, long-horizon AI product.

• Own work end-to-end: data preparation, training, evaluation, inference, and iteration.

• Turn research ideas into working systems that run reliably in production.

• Debug model failures and system issues using real production signals.

• Iterate quickly: ship, measure outcomes, refine, and repeat.

• Collaborate closely with research, product, and engineering to deliver real user impact.

• Mentor and review work from other ML engineers through example and technical judgment.

• Work under real production constraints: latency, cost, reliability, and safety

Tech Stack

• Python

• PyTorch / JAX

• GPU-based training and inference systems

Ideal Experience

• You have built and shipped ML systems used by real users.

• You understand how modern ML models behave — and misbehave — in production.

• You write strong, production-quality code and think in systems, not scripts.

• You take ownership, work independently, and push work across the finish line.

• You learn fast, communicate clearly, and improve through iteration.

How We Work

Our organization is very flat and our team is small, highly motivated, and focused on engineering and product excellence. All members are expected to be hands-on and to contribute directly to the company’s mission.

Interview process

If there appears to be a fit, we'll reach to schedule 3, but no more than 4 interviews.

Applications are evaluated by our technical team members. Interviews will be conducted via virtual meetings and/or onsite.

We value transparency and efficiency, so expect a prompt decision. If you've demonstrated the exceptional skills and mindset we're looking for, we'll extend an offer to join us. This isn't just a job offer; it's an invitation to be part of a team that's bringing AI to have practical benefits to billions globally.