Top Machine Learning Jobs in Austin, TX

Design and build AI-native, high-performance distributed backend platforms for IT/OT/cybersecurity. Integrate LLMs and ML into production workflows, develop data pipelines and real-time processing, mentor engineers, and drive architectural direction and AI-assisted engineering practices.

Lead design and delivery of AI/GenAI solutions and scalable data infrastructure. Drive business development, manage multiple client engagements, mentor teams, implement AI models using Python and C++, ensure data quality, and promote innovation across projects.

Technical lead for end-to-end ML and CV pipelines to reconstruct, validate, and maintain map primitives from multi-modal vehicle sensor data. Architect scalable offline and production map reconstruction systems, define technical strategy, drive cross-functional integration, mentor senior engineers, and ensure robustness, safety, and operational scalability for national-scale mapping.

Build and maintain Cloudflare's new vector database offering: design, implement, and operate distributed, low-latency cloud systems. Own features across the stack, ensure reliability and performance, write design documents, and apply machine learning and data retrieval expertise using TypeScript, Go, Rust, and related technologies.

Lead and grow a machine learning team to build predictive models and ML features that drive revenue, retention, and product decisions. Partner with product and engineering, conduct exploratory analysis, design experiments, mentor engineers, and communicate insights to stakeholders.

Partner with healthcare clients to discover, design, and deliver production-grade AI/LLM solutions. Architect cloud-native systems, build and deploy ML/LLM applications, prototype rapidly, troubleshoot distributed systems, and lead client engagements to drive measurable business outcomes while collaborating across product, platform, and engineering teams.

Lead product analytics to inform strategy and optimize user experience: translate business questions into analysis, build reliable analytics infrastructure, run A/B tests and ML/statistical analyses, track funnels and cohorts, support 0-to-1 product launches, and communicate actionable insights to cross-functional teams.

Lead and scale a cross-functional AI and engineering organization that builds production-grade AI models, automated data pipelines, and LLM-driven workflows to power Samsara's GTM operations. Define a multi-year AI roadmap, set quality and review standards for model outputs, collaborate with execs across Engineering, Product, and RevOps, and hire and mentor managers and individual contributors to drive adoption and reliable automation across CRM, telemetry, and enterprise data systems.

Manage and mentor a global team of data engineers and data scientists, align technical delivery with business goals, allocate resources, improve delivery processes, drive automation and technical evangelism, and collaborate with sales and account teams to support growth.

Lead AI engineering teams to design, build, and deploy end-to-end AI solutions. Manage client relationships, plan and deliver projects, mentor staff, drive adoption of cloud-native microservices, CI/CD, vector databases and orchestration tools, and ensure high standards of quality, innovation, and compliance.

Lead teams applying advanced data science and machine learning to large-scale claims, clinical, and member data to generate actionable insights that reduce cost of care and improve population health. Manage client relationships, mentor staff, ensure project delivery and quality, develop payer-focused analytics products, present findings to senior stakeholders, and promote continuous improvement and knowledge sharing.

Lead the development of fraud prediction models using machine learning, collaborating across teams to build, scale, and monitor models in production.

Lead a team of ML engineers to develop fraud detection models, defining strategies, driving model evolution, and ensuring effective integration with cross-functional teams.

The Research Engineer will collaborate with scientists to develop and validate machine learning models for speech technologies, manage scalable training systems, and design tools for accessibility to non-technical users.

The Staff ML Engineer will design and operate Samsara's ML platform, collaborating with teams to enhance ML features and improve safety outcomes. Responsibilities include overseeing system reliability, leading technical direction, and mentoring engineers.

Intern will own and deliver a scoped engineering project on Deepgram's voice platform (ASR/TTS/LLM/API/SDK), ship reviewed code to staging/production, use AI in development workflows, improve a component measurably, and present results with documentation.

Provide legal guidance on Rubrik's AI use and products, ensuring compliance with laws and contracts, drafting and negotiating AI terms, coordinating generative AI review processes, educating employees, and improving AI review controls and workflows across cross-functional teams.

The Staff Machine Learning Engineer will lead ML initiatives, develop ML systems, collaborate with cross-functional teams, and mentor junior engineers to optimize healthcare intake workflows using advanced machine learning techniques.

Lead the design and implementation of critical Safety AI initiatives, focusing on real-time perception systems for edge devices and optimizing models for production deployment in safety-critical environments.

The Senior Data Scientist will analyze data to drive product strategy, optimize performance, and support product launches, collaborating with cross-functional teams.

Partner with healthcare clients to design, prototype, and deploy production-grade AI/LLM solutions. Lead discovery, architecture, cloud integration, and delivery while troubleshooting distributed systems and driving measurable client outcomes.

Lead productionization and optimization of ML models for a global, serverless inference platform. Build benchmarking/evaluation tooling, improve inference performance (quantization, compilation, batching), integrate models with heterogeneous GPUs and accelerators, enhance deployment workflows and observability, and mentor engineers while driving ML production best practices.

Lead applied research and end-to-end development of production ML systems for a high-volume e-commerce marketplace. Prototype and deploy models, optimize auction and decisioning logic, improve performance, robustness, and interpretability, and collaborate with product and engineering. Provide technical leadership for a small specialist team, set engineering standards, manage delivery, and mitigate project risks.

Design, build, and deploy end-to-end ML solutions for ranking, recommendations, search, and personalization. Develop scalable data pipelines, monitoring, and evaluation frameworks; run A/B tests and collaborate with product, engineering, and analytics to drive data-driven decisions and reliable production models.

Lead applied AI/ML initiatives, design and deliver production-grade software, develop secure high-quality code, troubleshoot, automate remediation, evaluate vendor architectures, champion best practices across engineering, and drive adoption of AI/ML, NLP, LLMs and streaming technologies.