AI Engineer

Writer; United States; Hybrid; USD 140,000 - 210,000; Full time

Benefits offered by this job

Health plans

Parental leave

Hybrid culture

Stock options

401k plan

Learning stipend

PTO

Office spaces SF/NY/London

Job summary

Writer is seeking an experienced ML/AI engineer to design, deploy, and scale production AI systems. You will collaborate across teams, own end-to-end features, and apply deep learning, NLP, and large language models to real-world problems.

Proficiency in Python and frameworks such as PyTorch, TensorFlow, or JAX is required. We offer a hybrid work culture with in-office and remote collaboration, health benefits, stock options, a 401(k), learning stipends, generous PTO, and offices in SF, NY, and

Qualifications

• 5+ years building and deploying ML/AI systems in production.
• Strong problem-solving and a passion for tackling complex AI challenges.
• Proficiency in Python and ML frameworks PyTorch, TensorFlow or JAX.
• Collaborate across diverse teams and own your projects with impact.
• Write clean, efficient, and well-tested code for scalable systems.
• Solid ML fundamentals, DL architectures, and NLP/LLMs.

Responsibilities

• Design and deploy production ML/AI systems with reliability.
• Collaborate with product and data teams to deliver impactful features.
• Mentor peers and uphold coding standards and tests.

Skills

ML production

Python

PyTorch

TensorFlow

JAX

NLP/LLMs

Tools

PyTorch

TensorFlow

JAX

AWS

GCP

Azure

MLOps

Job description

• 5+ years of professional experience building and deploying machine learning or AI systems in a production environment
• Exceptional problem-solving skills and a passion for tackling complex technical challenges with creative AI solutions
• Proficiency in Python and experience with relevant AI/ML frameworks such as PyTorch, TensorFlow, or JAX
• You Connect by collaborating seamlessly across diverse teams, Challenge the status quo to innovate, and Own your projects with a commitment to impact and excellence
• Demonstrated ability to write clean, efficient, and well-tested code for scalable software systems
• A strong foundation in machine learning principles, deep learning architectures, and natural language processing, particularly with large language models
• Experience with cloud platforms (e.g., AWS, GCP, Azure) and MLOps tools for managing the AI lifecycle

Benefits

• Health: Selection of employer-covered medical plans, plus dental, vision, life insurance, and FSA.
• Family-friendly: Competitive parental leave policy, working parents can truly own their schedules.
• Hybrid culture: We’re built around a mix of in office collaboration and remote work when needed.
• Finance: Company stock options, plus a 401k plan with employer contributions.
• Growth: Learning and development stipend and generous PTO.
• SF/NY/London office space: Three beautiful sunny offices in three incredibly exciting cities.