Member of Technical Staff, Machine Learning

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

Job summary

A forward-thinking AI firm in New York is seeking a proactive ML Engineer to build and improve machine learning components and models. You will enhance production systems by debugging issues and implementing tests, working closely with senior engineers. Ideal candidates have strong foundations in ML, experience deploying models, and are eager to learn. This position offers an opportunity to work in a collaborative, hands-on environment focused on engineering excellence.

Qualifications

• Strong foundations in machine learning and modern neural architectures.
• Some hands-on experience training, fine-tuning, or deploying ML models.
• Curious, coachable, and eager to learn from real systems in production.

Responsibilities

• Build and improve ML components for data processing and model inference.
• Fine-tune models as part of production systems.
• Implement evaluation techniques to understand model behavior.

Skills

Machine learning foundations

Python

Production-quality code

Model evaluation and testing

Continuous improvement

Tools

PyTorch

JAX

GPU systems

Job description

About the Role

A1 is building a proactive AI system that understands context across conversations, plans actions, and carries work forward over time.

You will work on core ML systems under the guidance of senior engineers. This role is for builders who want to grow their judgment by shipping real ML systems in production.

Focus

• Build and improve ML components across data, training, evaluation, and inference.

• Fine-tune and adapt models as part of larger production systems.

• Implement evaluation and testing to understand model behavior.

• Help build and maintain data pipelines for real-world and synthetic data.

• Debug model issues, performance problems, and production incidents.

• Ship improvements iteratively and learn from real user feedback.

• Work closely with senior ML engineers and product teams.

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

Tech Stack

• Python

• PyTorch / JAX

• Production ML systems running on GPUs

Ideal Experience

• Strong foundations in machine learning and modern neural architectures.

• Some hands-on experience training, fine-tuning, or deploying ML models.

• Comfortable writing production-quality code and learning new tools quickly.

• Curious, coachable, and eager to learn from real systems in production.

• Able to work through ambiguity with guidance and grow ownership over time.

• Bias toward shipping, iteration, and continuous improvement.

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.