AI / ML Engineer

Neuron Factory; San Francisco (CA); On-site; USD 120,000 - 160,000; Full time

Job summary

A pioneering software company in San Francisco is seeking an experienced AI/ML Engineer to lead critical initiatives and shape core machine learning systems. As a founding engineer, you'll take ownership of the entire lifecycle, from model prototyping to deploying production-grade systems. The position promises significant impact and ownership, allowing you to influence product direction and drive innovation in the field of machine learning. Join a visionary team to redefine productivity with your expertise.

Qualifications

• 8+ years of experience in AI, ML, and Software Engineering.
• Strong foundation in ML/DL algorithms and statistical modeling.
• Experience with modern ML frameworks, particularly with LLMs.

Responsibilities

• Design, build, and deploy scalable ML systems for core product.
• Own end-to-end development from problem formulation to deployment.
• Collaborate with product and engineering for ML solutions.

Skills

Artificial Intelligence

Machine Learning

Software Engineering

ML/DL algorithms

Statistical modeling

Data processing

Python

APIs

Cloud infrastructure

Containerization

Education

Degree in Computer Science

Tools

PyTorch

TensorFlow

JAX

Airflow

MLflow

SageMaker

Vertex AI

Job description

About the Company

We’re a software company on a mission to build the AI Co-Worker, a powerful platform designed to learn complex jobs and deploy intelligent agents to solve them at Scale.
Our technology sits at the intersection of AI and human productivity, enabling businesses to automate sophisticated workflows like never before.

About the Role

As we scale, we are looking for an AI/ML Engineer to take ownership of critical initiatives, drive innovation, and help shape the future of our ML systems and products.

As our Founding AI/ML Engineer, you’ll be one of the first technical hires and play a key role in building our core machine learning systems from the ground up. You’ll own the entire lifecycle; from prototyping models to deploying production-grade systems while working closely with the founding team to influence product direction and technical architecture.

This is a rare opportunity to have deep impact, massive ownership, and the chance to shape the culture, systems, and trajectory of a category-defining company.

Requirements

• Design, build, and deploy scalable ML systems that power our core product.
• Own end-to-end development: problem formulation, data pipelines, model training, evaluation, and production deployment.
• Collaborate with product and engineering to translate business goals into ML solutions.
• Evaluate and apply state-of-the-art techniques in LLMs, deep learning, classical ML, and generative AI.
• Establish best practices for model reproducibility, monitoring, and performance tracking.
• Build and scale our MLOps infrastructure (training pipelines, experiment tracking, deployment tooling).
• Recruit and mentor future ML/AI hires and help grow the team and culture.

Qualifications

• 8+ years of experience in Artificial Intelligence, Machine Learning and Software Engineering.
• Strong foundation in ML/DL algorithms, statistical modeling, and data processing.
• Deep experience with modern ML/DL frameworks (e.g., PyTorch, TensorFlow, JAX).
• Hands-on experience with LLMs, transformers, or other foundation models.
• Proven experience building and deploying production-grade ML systems.
• Strong software engineering skills (Python, APIs, cloud infra, containerization).
• Familiarity with ML pipelines and platforms (e.g., Airflow, MLflow, SageMaker, Vertex AI).
• Startup mindset: ownership, bias for action, comfort with ambiguity, and a builder mentality.
• Degree in Computer Science, Machine Learning, Mathematics, or related field.

Desired Skills

Nice to Have: Computer Science PhD or MS preferred (but not required with equivalent experience).