Top Senior Machine Learning Jobs in Austin, TX

Build and ship production AI systems for enterprise portfolio and planning workflows. Develop anomaly and signal detection, LLM-powered insight synthesis, natural-language planning rule translation, evaluation frameworks, and MCP tools. Partner with external AI providers initially, then establish internal standards for prompt design, model selection, agent architecture, reliability, and evaluation. Collaborate closely with product, design, and domain engineering teams.

Designs and delivers end-to-end AI/ML solutions on Google Cloud, including data ingestion, feature engineering, model training, deployment, and monitoring. The role leads client discovery, translates business needs into AI/ML strategies, develops roadmaps, provides technical leadership, and supports testing and user training. It requires hands-on Gemini and GCP expertise, customer-facing consulting experience, and collaboration across engineering and product teams.

Leads Hightouch’s machine learning organization, including product development, technical roadmapping, recruiting, reliability, and people management. The role focuses on personalization, product recommendations, probabilistic identity resolution, LLM-powered content generation, and scalable machine learning infrastructure for model training and inference.

Design and build a production-grade AI memory and retrieval engine for personalized, privacy-preserving LLM experiences. Responsibilities include developing multi-tier storage and hybrid retrieval pipelines, memory extraction and classification, adaptive updates, evaluation frameworks, and reliability safeguards. The role researches advanced memory architectures, supports agentic workflows and tool integrations, collaborates with product management, and owns production operations. Strong software engineering, Python, NLP/ML, vector database, LLM orchestration, privacy, and RAG expertise are required.

Design and build production-grade AI memory systems, including retrieval pipelines, storage tiers, privacy filtering, memory extraction, adaptive updates, hybrid search, and evaluation frameworks. Ship and maintain LLM-powered features, diagnose RAG and prompt failures, implement agentic workflows and tool integrations, and ensure reliable degradation and fallback strategies. Collaborate with product management to define feasible memory-system capabilities while owning production reliability.

Develop and validate machine learning models, training pipelines, inference systems, and numerical methods using Python. Debug algorithmic and system-level issues, optimize memory usage and performance, design systematic tests, and build reproducible workflows. Review challenging ML engineering tasks, use coding agents in development, document technical findings, and deliver high-quality work independently within deadlines.

Develop and deploy machine-learned interatomic potentials for PFAS substitution. Responsibilities include generating and curating first-principles training data, building reproducible cloud and HPC workflows, training and validating ML force fields, running molecular dynamics and quantum chemistry campaigns, predicting physical properties, extending models to condensed-phase and process chemistries, and communicating model uncertainty and domain limitations.

Lead design, development, and production deployment of ML and algorithmic solutions. Own end-to-end ML systems, support distributed production reliability, communicate tradeoffs, and apply generative AI, LLM fine-tuning, RAG, and agentic techniques to product problems.

Build and improve large-scale machine learning models for aviation prediction products, including runway landing times, gate arrivals, and taxi durations. Develop high-performance systems using historical aviation data and complex data types, contribute ideas in an agile environment, and improve operational efficiency for airlines and airports. Preferred experience includes Kubernetes, AWS, GPU acceleration, graph neural networks, geospatial data, observability, ETL orchestration, and large-scale data processing.

Own the full lifecycle of production classification and detection models, including transformer fine-tuning, entity detection, behavioral risk scoring, training, serving, versioning, evaluation, drift detection, and retraining. Build SageMaker-based ML systems and define output contracts supporting audit and policy enforcement. Lead the evaluation harness and mentor a co-op while operating as the company’s sole ML engineer.

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.

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.

Designs, deploys, and operates production machine learning systems for identity resolution, scoring, deliverability, and personalization. Responsibilities include feature engineering, model training, real-time inference, graph database integration, monitoring, drift detection, retraining, low-latency APIs, and scalable data processing. The role also contributes to MLOps standards, GPU acceleration, and ML architecture while partnering with platform engineering.

Lead the design, development, deployment, and operation of production-grade generative AI and LLM solutions. Build data and ML pipelines, fine-tune and evaluate models, implement RAG and prompt engineering, and establish MLOps practices on AWS. Drive use cases across insurance workflows, ensure security and responsible AI compliance, mentor engineers, and collaborate with product, engineering, security, finance, and go-to-market teams.

Develop autonomy evaluation metrics, statistical and machine-learning analyses, dashboards, and pipelines for simulation and on-road autonomous vehicle testing. Analyze perception, prediction, and planning performance; identify anomalous behavior and critical scenarios; apply VLMs and LLMs where appropriate; and support release gating and safety decisions. The role requires production Python, C++ debugging, technical leadership, cross-functional collaboration, and rigorous software engineering practices.

Develop and deploy machine learning models for fraud detection and credit risk decisioning. Own the full ML lifecycle, including data exploration, feature engineering, experimentation, validation, deployment, monitoring, and optimization. Analyze fraud trends and portfolio outcomes, collaborate with Product, Engineering, and Risk Operations, apply generative AI tools, and mentor fellow scientists while delivering measurable improvements in fraud prevention and responsible credit access.

Build and operate distributed GPU training platforms for foundation models. Run and debug multi-node PyTorch and Ray Train workloads, tune GPU networking and scheduling, profile performance, and develop benchmarks and evaluation harnesses. Partner with infrastructure teams and client scientists to improve throughput, reliability, cost efficiency, observability, and end-to-end training pipelines on GKE.

Design, build, and deploy production machine learning systems across workflow automation, data enrichment, predictive insights, and agent-based workflows. Own the full ML lifecycle, including data ingestion, model training, evaluation, deployment, and feedback loops. Integrate AI into backend services, APIs, data pipelines, and customer-facing products while shaping architecture and long-term AI strategy. Collaborate with product, engineering, and customers to deliver scalable, reliable AI solutions.

Develop and deploy machine learning systems for humanoid robot manipulation, covering data collection, annotation, synthetic data generation, model training, simulation, hardware deployment, and fine-tuning. Diagnose vision, calibration, sensor fusion, model, and data bottlenecks to improve reliability. Build real-time robotic software using modern computer vision and deep learning, while mentoring engineers and driving architectural excellence across the robotics organization.

Lead the development and production deployment of applied machine learning and LLM solutions across product, marketing, and operations. Own ranking, recommendation, filtering, personalization, experimentation, and model evaluation systems. Partner with business and technical teams to identify valuable use cases, establish modeling best practices, and ensure reliable production deployment. Manage and mentor applied ML practitioners while maintaining technical rigor, business alignment, and measurable customer and revenue impact.

Lead the design, development, deployment, and operation of production AI and ML systems. Build high-throughput Python services and Databricks pipelines, support traditional ML and LLM applications, establish testing and deployment standards, and develop monitoring for quality, reliability, latency, throughput, and cost. Partner with product, research, and responsible AI teams on governance and release criteria. Provide architectural leadership, conduct code reviews, mentor engineers, and resolve complex technical dependencies.

Own the end-to-end machine learning lifecycle, including data preparation, dbt pipelines, feature engineering, model development, deployment, monitoring, and evaluation. Apply ML to risk assessment, cost prediction, image analysis, and business operations. Contribute to architecture and strategy while partnering with Engineering, Product, Operations, and Business teams. Build scalable solutions using Python, AWS, Kafka, Django or FastAPI, PostgreSQL, and GitHub Actions.

Leads Tala’s Machine Learning Engineering team in building and operating platforms for model development, real-time inference, streaming features, batch processing, deployment, monitoring, and production reliability. Manages and develops 4–6 engineers, owns delivery and prioritization, provides architectural leadership, improves engineering practices and SLOs, and partners with Data Science, Data Engineering, Product, Credit, and Business Development teams.

Build and deploy scalable AI and machine learning products for clinical research, including patient recruitment, enrollment forecasting, and study feasibility. Develop ML models, LLM-powered agents, data pipelines, and MLOps infrastructure using cloud services and CI/CD automation. Collaborate with cross-functional teams to integrate AI into clinical research workflows, monitor model performance, and advance reliable, unbiased solutions that improve patient outcomes.