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.

Design and deliver enterprise-grade AI workflow transformations on GCP. Build secure, event-driven backend services and multi-step AI agents, integrate bots with enterprise systems, implement telemetry, testing, CI/CD, HITL validation, and create reusable architecture, documentation, and runbooks while guiding adoption and governance.

Design, build, and scale AI-enabled workflow and automation solutions across collaboration platforms and enterprise systems. Translate manual processes into secure, reusable, event-driven automations using Python/JavaScript, integrate AI capabilities like Gemini, implement logging and telemetry, and partner with product, security, and business teams to deploy and sustain enterprise transformation solutions.

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.

Lead development of scalable, reliable continuous integration infrastructure supporting autonomous-vehicle development, machine learning training, simulation, and remote builds. Specialize in Remote Build Execution and a FUSE-based file system, enabling source-code editing and developer workflows. Design and implement productivity improvements, evaluate technologies, influence technical roadmaps, establish engineering best practices, manage technical debt, and mentor engineers while balancing business and customer priorities.

Design, build, and deploy scalable ML training and evaluation platforms for autonomous driving. Lead architecture and implementation of distributed, high-performance pipelines, drive cross-team prioritization, mentor engineers, and support recruiting and code quality to accelerate ML model development lifecycle.

Lead the development and production deployment of machine learning, generative AI, and multi-agent solutions. Translate business problems into scalable AI products, build models and data pipelines, architect cloud and lakehouse systems, and implement MLOps, LLMOps, evaluation, governance, security, and monitoring. Partner with cross-functional stakeholders, communicate technical outcomes, troubleshoot production issues, mentor teammates, and shape the strategic roadmap for enterprise AI capabilities across Azure, Databricks, AWS, and GCP.

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 and development of scalable, high-performance ML training infrastructure. Drive distributed training performance optimization, observability, and developer experience. Own cross-functional infrastructure initiatives, set technical direction and standards, and mentor engineers to deliver platform capabilities that support large-scale model training.