Top AI & Machine Learning Jobs in Austin, TX

Leads and develops a team of AI Customer Success Architects delivering enterprise Rovo engagements from implementation through production value. Oversees capacity planning, engagement quality, adoption metrics, customer outcomes, and team growth. Partners with sales, customer success, product, and engineering teams to prioritize engagements, improve delivery methods, surface product feedback, and support AI adoption. The role also involves executive customer relationships, internal and external advocacy, regulatory awareness, and up to 15% travel.

Develop and deploy scalable machine-learning solutions, particularly deep-learning forecasting models for company top-line metrics. Lead the full ML lifecycle from problem definition and experimentation through production deployment and improvement. Design production-grade architectures, evaluate and improve models, collaborate with cross-functional teams, explain technical concepts to varied audiences, and mentor junior engineers.

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.

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.