Top AI & Machine Learning Jobs in New York City, NY

Principal Machine Learning Engineer responsible for setting technical direction and building production-grade AI systems for Confluence. The role spans model evaluation, retrieval, ranking, prompt and workflow design, experimentation, reliability, and customer-facing AI product development. Responsibilities include making architectural decisions, partnering across engineering and product teams, mentoring engineers, identifying model and product failure modes, and delivering scalable AI experiences across content creation, editing, discovery, recommendations, and multimodal interaction.

Design, build, and operate scalable, low-latency search and retrieval systems powering Atlassian products. Develop and serve production ML models (neural rankers, embeddings, rerankers) using PyTorch and Triton. Implement vector indexing and RAG/agentic retrieval pipelines, ensure observability and cost optimization, manage on-call/runbooks, and collaborate cross-functionally while mentoring junior engineers.

Develop and deploy machine learning and LLM capabilities for agentic search, query understanding, search planning, and retrieval. Build evaluation datasets, improve model behavior through experimentation and fine-tuning, and optimize quality, latency, cost, and reliability. Collaborate across search, product, infrastructure, and AI teams while owning projects through production and mentoring engineers.

Lead the technical direction for Atlassian’s Search Serving platform, building globally scalable, low-latency agentic search systems. Own architecture for retrieval, indexing, query processing, and ML inference while improving quality, reliability, scalability, operability, and cost. Establish production ML practices for evaluation, experimentation, observability, rollouts, and incident response. Optimize inference for embedding, retrieval, and ranking models, and lead cross-team initiatives while mentoring senior engineers and aligning technical leaders.

Build and maintain core infrastructure for developing, training, evaluating, deploying, and operating machine learning models and pipelines for Atlassian’s GenAI products. Solve complex software infrastructure and architecture challenges, lead engineers through technical projects from design to launch, and collaborate with internal teams and customers. The role involves large-scale distributed systems, data processing, cloud platforms, CI/CD, and machine learning lifecycle support.