Top Machine Learning Engineer Jobs in Austin, TX

Design and implement AI agents for mission planning and orchestration across autonomous systems, focusing on LLM frameworks and edge computing.

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.

Develop machine learning capabilities, custom analytics, and mission-critical AI solutions across image, video, text, geospatial, time-series, and structured data. Build and orchestrate data engineering pipelines for agentic and traditional architectures, collaborate with data scientists, software and DevOps engineers, contribute to the Chariot AI operations platform, and support customer-facing deployments.

Build scalable systems and machine learning models that transform satellite imagery into labeled geospatial data and production computer vision products. Responsibilities include developing data and annotation pipelines, tracking lineage and versioning, deploying inference services, evaluating models, supporting cloud infrastructure, and improving labeling workflows. The role requires strong Python, cloud, containerization, CI/CD, infrastructure-as-code, production support, and deep learning expertise, with preferred experience in GCP, Vertex AI, spatial databases, and satellite imagery.

Design, implement, benchmark, and optimize CUDA-based GPU kernels and custom operators for on-vehicle ML inference. Build tooling for profiling and debugging, integrate kernels into the ML runtime, collaborate with compilers and architecture teams, and deliver high-performance, production-grade libraries meeting strict latency, throughput, and reliability targets.

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.

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.

Own the strategy, architecture, development, deployment, monitoring, and continuous improvement of production personalization and recommendation systems. Build custom embeddings, ranking models, and shared representations across feeds, search, and recommendations. Lead ambiguous problems from concept to measurable customer impact, establish AI evaluation standards, partner with product and data teams, make cross-team technical decisions, and coach senior engineers.

Own and evolve Braze’s ML platform for globally scaled training pipelines, model serving, deployment, orchestration, and observability. Lead complex infrastructure initiatives, improve reliability and cost efficiency, manage incident response, and establish production quality standards. Partner across teams, shape technical strategy, mentor senior engineers and data scientists, and connect platform decisions to customer and business outcomes.

Design and productionize ML models and agentic systems for intraday financial alpha generation. Build data pipelines, experiment frameworks, and portfolio optimization methods. Collaborate with quants and engineers to move models from research to live trading environments.

Lead the architecture, deployment, and scaling of agentic AI infrastructure for embedded memory design, characterization, DFT, silicon bring-up, and yield optimization. Develop multi-agent LLM workflows, integrate AI systems with EDA tools and enterprise data, automate scan-failure analysis, build predictive yield models, and establish evaluation, testing, and safety guardrails. Collaborate with memory designers, DFT experts, EDA specialists, and infrastructure engineers to deliver production-grade AI engineering solutions.

Build and ship production AI systems for enterprise portfolio and planning workflows. Develop anomaly and signal detection, LLM-powered insight synthesis, natural-language planning rule translation, evaluation frameworks, and MCP tools. Partner with external AI providers initially, then establish internal standards for prompt design, model selection, agent architecture, reliability, and evaluation. Collaborate closely with product, design, and domain engineering teams.

Designs, deploys, and operates production machine learning systems for identity resolution, scoring, deliverability, and personalization. Responsibilities include feature engineering, model training, real-time inference, graph database integration, monitoring, drift detection, retraining, low-latency APIs, and scalable data processing. The role also contributes to MLOps standards, GPU acceleration, and ML architecture while partnering with platform engineering.

Lead the technical direction of Quora’s recommendation ecosystem, including feed retrieval, ranking, re-ranking, optimization, data pipelines, feature engineering, model training, and production integration. Drive measurable improvements through experimentation, apply generative AI, mentor engineers, and partner across product, data science, and infrastructure teams.

Develop, implement, and deploy scalable machine learning models for healthcare solutions. Collaborate cross-functionally to integrate models, analyze data to improve performance, ensure data quality, and stay current with ML advancements to support patient care and operational efficiency.

Develop and deploy reliable machine learning, LLM-powered agent, and agentic workflow systems for government and enterprise customers. Integrate models with tools, APIs, MCP servers, and data sources; build evaluation frameworks for performance, reliability, and safety; improve Chariot, the company’s AI operations platform; and support mission-critical deployments. Collaborate with data scientists, software engineers, DevOps teams, and customers in secure government environments.

Develop and deploy reliable machine learning, LLM, agentic, and data-centric systems for government and enterprise customers. Build and evaluate AI agents, workflows, tool integrations, and retrieval-augmented systems; measure performance, reliability, and safety; and improve the Chariot AI operations platform. Collaborate with customers, data scientists, software engineers, and DevOps teams on mission-critical deployments. The role is fully remote or hybrid in Austin, with travel up to 20%.

Own the end-to-end computer vision platform for fine-grained item identification and pricing. Responsibilities include designing and fine-tuning vision models, defining evaluation and confidence standards, building labeling feedback loops, prioritizing coverage versus accuracy, and deploying reliable, low-latency serving infrastructure at scale. The role also communicates model capabilities and trade-offs to technical and non-technical stakeholders.

Develop and deploy machine learning systems for humanoid robot manipulation, covering data collection, annotation, synthetic data generation, model training, simulation, hardware deployment, and fine-tuning. Diagnose vision, calibration, sensor fusion, model, and data bottlenecks to improve reliability. Build real-time robotic software using modern computer vision and deep learning, while mentoring engineers and driving architectural excellence across the robotics organization.

Design, develop, and implement AI/ML models; analyze large datasets; integrate AI solutions into existing systems; apply MavLink for autonomous systems; collaborate across teams and improve ML development processes.

Design, build, and deploy autonomous AI agents using LLMs and agent frameworks; integrate agents into e-commerce and supply chain systems; implement multi-agent orchestration, memory and tool integrations; fine-tune models and apply RL concepts; and build safety, monitoring, and MLOps pipelines.

Develop and maintain production machine learning models, build scalable training and serving technologies, optimize real-time and batch ML pipelines, and monitor model performance and drift. The role involves deep neural networks, recommendation systems, automated retraining, ML research, engineering excellence, and communicating technical concepts to diverse audiences. Candidates need 6+ years of industry ML experience, strong coding skills, and experience deploying neural networks at scale.

Lead the design, productionization, and operation of machine learning and agentic AI capabilities. Partner with product engineering, shape AI/ML architecture, build evaluation systems and data pipelines, and develop reliable MLOps tooling. Champion responsible AI, safety, latency, cost, and error handling. Serve as a technical resource for engineers, influence direction, mentor others, and occasionally explain AI functionality to customers or customer-facing teams.