Top Machine Learning Jobs in Austin, TX

Lead GitLab’s company-wide strategy for autonomous, agentic software development workflows. Identify and validate AI opportunities across the SDLC, build prototypes and reference architectures, establish evaluation frameworks and safety controls, and drive internal adoption. Partner across Engineering, Product, Infrastructure, Security, Architecture, and Data and ML teams to scale reliable, observable, compliant agentic capabilities into customer-facing products. Mentor senior technical leaders and represent GitLab externally on AI-assisted development and productivity measurement.

Define and lead AI/ML architectures for embedded, edge, cloud-connected, and intelligent sensor solutions. Develop reusable AI frameworks, inference pipelines, deployment methods, and reference implementations. Evaluate machine learning, deep learning, foundation models, and agentic AI while optimizing accuracy, latency, memory, power, and cost. Partner with engineering teams, product groups, research organizations, and strategic customers, provide technical mentorship, shape AI roadmaps, and contribute to technical publications, patents, and ecosystem initiatives.

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.

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.

Develop AI-native industrial automation systems for real-time reasoning, optimization, autonomous operation, and continuous learning at the edge. Design distributed architectures combining generative AI, machine learning, reinforcement learning, streaming analytics, and control systems. Deploy low-latency AI on edge hardware and integrate it with robotics, sensors, controllers, and industrial infrastructure while collaborating with AI research, controls, embedded systems, and operations teams.

Owns end-to-end architecture for a major generative AI platform and shared AI services. Responsibilities include designing model integrations, APIs, data flows, orchestration, RAG and agentic workflows; defining architecture standards, integration contracts, and non-functional requirements; ensuring security, resilience, compliance, governance, observability, and scalability; and conducting architecture reviews and proof-of-concept assessments. The role partners with product, engineering, data science, cloud, security, and control teams to deliver production-ready enterprise AI systems.

Own applied data science for content performance, audience behavior, attribution, segmentation, and pilot performance prediction. Build models for engagement, sentiment, conversion, retention, and campaign attribution; track forecast accuracy and calibration; and translate findings into actionable recommendations. The role also mentors a junior analyst, strengthens statistical rigor, and partners with research and data science leadership.

Develops, configures, tests, and supports AI solutions, workflows, integrations, and model-enabled applications. Creates technical instructions and documentation, troubleshoots and optimizes AI systems, coordinates technical tasks, and collaborates with architects, delivery leaders, project teams, and stakeholders. Requires a college degree or equivalent, three years of related experience, technical communication skills, and the ability to travel and participate in on-call availability.

Leads enterprise AI architecture and strategy across the university’s digital ecosystem. Designs scalable, secure, ethical AI, machine learning, automation, and analytics solutions integrated with enterprise platforms and data systems. Establishes AI governance, responsible AI standards, MLOps/LLMOps practices, monitoring, lifecycle management, and security controls. Partners with business, product, data, engineering, cybersecurity, and platform teams; evaluates models and vendors; guides implementations, workshops, proofs of concept, adoption initiatives, and team development.

Review and annotate robotic-system video footage by identifying objects, movements, events, visual features, and anomalies. Provide feedback to improve video datasets and annotation consistency, collaborate with AI teams on annotation protocols, document observations, participate in quality reviews, and recommend improvements to video capture and labeling strategies.

Analyze production drive-thru voice AI data, build SQL reports and dashboards, monitor KPIs, identify conversation error patterns, and improve NLU models, conversation flows, and LLM behavior through machine learning and rule-based techniques.

Oversee global analytics and algorithm design initiatives for a nonprofit: identify high-impact projects, coordinate with student directors, collect feedback, design and run training for developers and technical advisors, and form partnerships to support implementation. Volunteer commitment ~5-6 hours/week.

Lead data science projects from ideation through enterprise impact, building scalable and well-tested solutions that improve healthcare outcomes. Apply probability, statistics, machine learning, SQL, and Python to develop data-driven products and insights. Partner with stakeholders, assess product performance, identify strategic improvements, mitigate risks, and mentor teammates while aligning initiatives with business objectives.

Build and productionize traditional machine learning models, owning feature engineering, validation, deployment, monitoring, drift detection, and retraining. Develop LLM-powered agentic tools, dashboards, self-serve analytics products, and automated insight pipelines. Partner with engineering and business stakeholders, write production-grade Python and SQL, drive adoption, improve data quality, and establish responsible AI workflows for a regulated industry.

Lead ecosystem-level data science and econometric analyses across Xbox, Game Pass, Console, PC, and Store experiences. Evaluate growth, engagement, monetization, experiments, incrementality, and investment ROI using statistical and causal inference methods. Build reusable analytical products and self-service capabilities, influence executive decisions, and mentor team members.

Supports the design, validation, and operationalization of predictive, machine learning, and AI models. Analyzes structured and unstructured data, develops algorithms, validates outcomes, creates visualizations, and communicates findings to stakeholders. Collaborates cross-functionally to integrate solutions and prepares model pipelines for business use. The role also involves statistical analysis, NLP, exploratory data analysis, feature engineering, and research into emerging technologies.

Analyze large datasets, identify meaningful patterns, develop machine learning models, and deliver actionable insights for business decisions. Collaborate with cross-functional teams to understand data needs, communicate findings clearly to stakeholders, and stay current with advances in data science and AI technologies.

Develop and deploy AI solutions from problem framing through production. Responsibilities include identifying AI opportunities, leading MVP development, building data pipelines, experimenting with generative LLMs, fine-tuning NLP models, and delivering scalable features. The intern will collaborate cross-functionally, communicate technical impact to business audiences, and work with cloud platforms such as AWS or Google Cloud.

Lead end-to-end LLM inference performance across serving systems, models, GPU hardware, and workloads. Define rigorous benchmarks, diagnose bottlenecks from scheduling through kernels and interconnects, optimize single- and multi-node deployments, and deliver production-ready runtimes and configurations. Collaborate with product and infrastructure teams, evaluate inference technologies, and implement runtime fixes. The role requires strong Python engineering, production experience with serving engines such as vLLM or SGLang, GPU profiling, and knowledge of modern inference optimization techniques.

Build and deploy production machine learning systems that optimize Samsara’s global hardware supply chain. Responsibilities include demand, inventory, lead-time, spend, and supplier-risk forecasting; anomaly detection; feature engineering and ETL; real-time inference; MLOps; data infrastructure improvements; governance; and model monitoring. The role defines the AI transformation roadmap, partners with executive and cross-functional stakeholders, mentors scientists, and drives adoption of AI-enabled operational workflows.

Lead client engagements from data onboarding through production, plan studies that demonstrate business value, and build automated machine learning workflows using Python, scikit-learn, pandas, and Spark. Translate client business questions into data solutions, manage stakeholder relationships, evaluate model performance, and inform product planning. Requires healthcare or life sciences data experience and end-to-end delivery of sophisticated data or machine learning solutions.

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.

Design, build, evaluate, and productionize machine learning and LLM-based agentic systems for healthcare products. Own projects end-to-end, including problem framing, dataset development, modeling, evaluation, deployment, monitoring, and safety guardrails. Build ML infrastructure, data pipelines, inference systems, and evaluation frameworks. Collaborate with clinicians, product managers, and engineers, while setting technical direction and mentoring team members.

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.

Manages multiple data science and machine learning projects from scoping through deployment and continuous improvement. Coordinates Data Science, Engineering, Product, and business stakeholders; maintains plans, dependencies, risks, and reporting; facilitates Agile ceremonies; removes blockers; and supports MLOps readiness, model releases, monitoring, retraining, documentation, and process improvement.