Machine Learning Engineer (Generalist) — AI Lab
NYC - 4 days in office
$1.5 million (Base + Bonus + LTI)
About the lab
Join a small, elite AI lab tackling high-stakes problems with real-world impact. The team blends startup speed with serious resources. Engineers and researchers work side by side, moving fast, giving direct feedback, and shipping systems that matter. PyTorch has become the framework of choice for cutting-edge AI research and production systems. Companies like Tesla, OpenAI, and Meta rely on PyTorch engineers to build everything from autonomous driving systems to large language models. Your PyTorch expertise positions you at the forefront of AI innovation, working on problems that didn't exist five years ago.
What you’ll do
-
Own the full ML lifecycle from idea to production: data prep, training, diagnostics, iteration, deployment, and monitoring
-
Collaborate closely with research scientists to choose methods, design experiments, and interpret results
-
Build reliable, maintainable production code in Python and modern ML tooling
-
Improve workflows and contribute to scaling the lab from 0?1?100
What you’ll bring
-
Strong software engineering foundation and hands-on coding in Python
-
Fluency with ML tooling such as NumPy, Pandas, and PyTorch
-
Track record of shipping ML systems end-to-end, not just running prebuilt pipelines
-
Clear thinking, bias to action, and comfort in a candid, fast-moving environment
Nice to have
-
Experience optimizing training loops or performance in PyTorch
-
Breadth across classical ML and deep learning
-
Prior experience in small, high-ownership teams
Location
NYC. Hybrid collaboration with onsite time for key moments. Exact expectations are tailored to the team’s needs.
Compensation
Top-of-market package with base, bonus, and a long-term incentive aligned to the lab’s success.
Apply
If you’re a builder who loves owning ML systems end-to-end and working with exceptional peers, we’d love to meet you.
