ML Engineer at fast-growing AI startup

Join a high-caliber team in New York City focused on pushing the boundaries of machine learning. You will lead the design and training of advanced deep learning systems, leveraging your expertise in PyTorch or JAX. This role offers the opportunity to apply broad ML literature knowledge to build innovative, production-grade models that define the next generation of AI technology.

Fast-growing AI startup

• Design, train, and evaluate sophisticated deep learning systems using modern optimization techniques and specialized tooling.
• Implement and scale models using PyTorch or JAX to solve high-impact machine learning problems.
• Stay current with and apply the latest research from ML literature to enhance system performance and architecture.

• At least 2 years of experience at top tech companies or startups specifically training Large Language Models (LLMs).
• Deep technical knowledge of modern deep learning architectures, optimization strategies, and the broader machine learning landscape.
• Proven proficiency in PyTorch or JAX with a track record of deploying robust ML systems into production.

• Work alongside elite engineering talent from top-tier technology companies and research institutions in a collaborative NYC environment.
• Join a well-funded startup backed by top-tier VCs at a critical stage of architectural development and model scaling.
• Tackle complex challenges that go beyond simple generative models, impacting the future of modern deep learning architectures.