Top Senior Machine Learning Jobs in Austin, TX

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.

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.

Leads a team developing AI capabilities for CrowdStrike’s cross-platform endpoint sensor. Owns the AI-on-endpoint roadmap, including machine learning model operationalization, telemetry data pipelines, on-device inference, architecture, feature delivery, and cross-functional collaboration. Manages hiring, mentoring, engineering quality, testing, performance, and releases across Windows, macOS, and Linux sensor platforms.

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.

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.

Create, solve, review, and validate coding-heavy machine learning tasks for AI training. Build and optimize models, training and inference workflows, data pipelines, evaluation systems, and reproducible Python workflows. Diagnose numerical and system-level failures, improve memory use and throughput, and verify technical correctness while working with coding agents.

Lead AI/ML initiatives for Air Force condition-based maintenance, from predictive model development through production deployment. Architect data pipelines, develop machine learning, deep learning, NLP, and generative AI solutions, and build knowledge graph capabilities. Provide technical leadership, integrate models into applications and workflows, and monitor data quality, model performance, drift, and bias. Collaborate with engineering, cloud, software, and product teams while supporting defense aviation logistics and sustainment missions.

Design and build large-scale mapping algorithms and geospatial data systems for automated map reconstruction and maintenance. Develop lane, road-network, geometry, topology, and map-conflation systems; process sensor and third-party data through distributed pipelines; apply machine learning and computer vision for feature extraction and change detection; and create automated validation and quality systems. Partner with perception, localization, simulation, and platform teams while mentoring engineers and driving technical problems into production.

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.

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 and operate production ML and AI systems powering patient experiences such as matching, ranking, recommendations, onboarding, and personalization. Build reusable ML infrastructure, pipelines, evaluation, observability, and model-serving capabilities. Provide technical leadership across the ML lifecycle, guide architecture, evaluate custom versus foundation-model approaches, and partner with ML and Patient Engineering teams to deliver reliable, scalable AI products.

Own and advance Wonderlic’s AI-powered Jobs Engine, using production machine learning and NLP to analyze labor-market data and infer job characteristics at scale. Partner with product, engineering, and I-O psychology teams to integrate responsible AI into assessment and workforce solutions. Apply fairness, explainability, legal defensibility, and rigorous evaluation practices to employment-related AI systems, including generative AI outputs and assessment interpretation.

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%.

Leads the AI Engineering team across applied AI features and foundational ML infrastructure. Responsibilities include managing Senior and Staff engineers, hiring and coaching, deploying LLM-powered capabilities, building MLOps and AI observability platforms, balancing product delivery with platform investment, and ensuring privacy, compliance, evaluation rigor, and clinical safety in production systems.

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.

Designs, trains, evaluates, and operationalizes intelligent document-processing solutions using OCR, machine learning, LLMs, Python, and cloud data pipelines. Builds extraction, validation, reconciliation, exception-handling, and human-review workflows for sensitive information. Develops secure, auditable, configuration-driven systems with API integrations, monitoring, testing, CI/CD, and infrastructure as code. Partners with business experts to define document fields, quality thresholds, validation rules, and acceptance criteria.

Leads client-facing discovery and technical product development for AI initiatives. Designs, prototypes, deploys, and operates agentic systems, retrieval pipelines, enterprise integrations, and production-grade generative AI applications. Establishes evaluation, observability, safety, governance, and reliability frameworks while translating business needs into technical specifications. Requires strong Python and software engineering expertise, cloud and DevOps experience, retrieval architecture knowledge, and at least seven years of engineering experience, including production generative AI experience.

Builds and deploys AI-enabled applications, machine learning models, data pipelines, APIs, and decision-support tools for aggregate and asphalt operations. The role optimizes plant automation, production efficiency, uptime, quality, pricing, sales processes, and costs using operational, equipment, quality, GIS, and sales data. It integrates AI with plant systems and legacy software, collaborates across operations and IT, and ensures responsible AI, cybersecurity compliance, documentation, and user adoption.

Design, build, and support high-throughput, low-latency LLM inference and data pipelines; collaborate with data scientists to post-train and evaluate models; implement scalable, production-ready systems with strong testing, monitoring, and performance focus; mentor engineers and iterate on architecture, reliability, and user-facing AI applications.

Own and improve Quora’s second-phase ranking models for feed and digest experiences. Build end-to-end machine learning systems covering data pipelines, candidate extraction, feature engineering, model training, and production integration. Identify new product applications for machine learning, develop efficient algorithms with other ML engineers, and contribute methodologies for large-scale recommendation systems.

Build and operate production machine learning systems for Safety AI, including low-latency APIs, data pipelines, model serving, evaluation, monitoring, and rollout infrastructure. Process large-scale camera and telematics data, optimize cloud and edge-to-cloud execution, track model performance and drift, and partner with applied scientists, firmware engineers, platform teams, and product managers to deliver reliable safety features.

Develop and productize machine learning models for autonomous vehicle perception, including LiDAR and camera perception, sensor fusion, detection, classification, segmentation, and tracking. Build scalable data, training, metrics, and performance workflows, and implement efficient C++ code for the NVIDIA DRIVE AV platform. Collaborate across engineering teams, track advances in machine learning, and optimize models for real-time embedded inference.