Top AI & Machine Learning Jobs in Austin, TX

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.

Leads AI/ML and data science products that improve engineering productivity. Translates ambiguous business challenges into scalable, production-grade solutions, shapes scientific direction, ensures methodological quality, mentors scientists, develops reusable frameworks, manages risks and delivery, and collaborates across engineering, data, software, and business teams. Communicates technical tradeoffs to stakeholders and leaders while driving measurable enterprise impact.

Develop data-intensive and AI-enabled applications supporting product development engineering teams. Build optimized SQL solutions, Databricks pipelines, Java and Python backend services, APIs, and cloud-native applications. Partner with AI teams to productionize ML and LLM solutions, improve reliability and security, troubleshoot production issues, and implement engineering best practices. The role also involves architecture, Agile collaboration, mentoring engineers, technical documentation, and driving modernization initiatives.

Lead architecture and hands-on development of reusable customer-facing AI platform services, including conversational AI, agent orchestration, retrieval, grounding, memory, tool integration, and policy enforcement. Define APIs, data models, deployment designs, testing, evaluation, and operational requirements. Build production systems across GCP and Azure while driving reliability, security, observability, CI/CD, and technical standards. Provide staff-level influence through architecture reviews, mentoring, code reviews, and cross-functional leadership without direct reports.

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.

Lead AI, machine learning, and data science workstreams from methodology through production delivery. Develop scalable, maintainable analytical code, explore new methods, influence product decisions, and collaborate with engineering, product, business, and data teams. Translate technical tradeoffs into actionable recommendations, present findings to technical and business audiences, and deliver measurable business impact while managing risks and maintaining methodological quality.

Build scalable evaluation, introspection, visualization, and metrics tools for autonomous vehicle machine learning models. Develop workflows processing billions of examples and connect offline evaluation with data, training, simulation, and launch decisions. Collaborate with modeling, data, infrastructure, and validation teams to improve model performance, safety, reliability, and development speed. Responsibilities include software design, implementation, maintenance, code reviews, and production tooling across backend, frontend, and data infrastructure.

As a Staff AI/ML Engineer, you'll enhance developer productivity, mentor engineers, drive best practices, and implement scalable solutions for AV tools and services.

Lead design, build, and operate scalable observability systems for distributed cloud and on-vehicle environments. Contribute code and reviews, create technical designs, build tooling to collect observability signals, run PoCs for new tools, partner with engineering teams, define SLIs/SLOs, guide roadmap decisions, and mentor engineers to improve developer productivity and reliability.

Lead design and delivery of production-grade AI capabilities to improve developer productivity across engineering tools, CI/CD, and embedded software workflows. Define technical vision, integrate AI into toolchains, drive productionization, establish responsible AI standards, mentor engineers, and influence cross-functional roadmaps and implementations.