Top Machine Learning Engineer Jobs in Austin, TX

Lead design and build of a scalable, multi-tenant AI/ML platform powering traditional ML, LLMs, and agentic systems. Own end-to-end architecture, production-grade backends, orchestration, LLMOps, vector DB integration, and cross-functional delivery while enforcing engineering standards, security, and observability.

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, build, and operate CV/ML infrastructure: manage a 130+ GPU bare-metal Kubernetes cluster with AWS bursting, own CV/ML CI/CD, maintain model/data/versioning and experiment tracking, automate training/testing, implement model optimization for edge (Jetson/TensorRT/quantization), and provide ML observability and security for production deployment.

The Staff Machine Learning Engineer will design and build scalable ML systems, mentor engineers, and drive AI strategy at SailPoint, focusing on AI-powered identity security solutions.

Lead technical strategy and build production-grade recommendation, ranking, and personalization ML systems. Design, deploy, and scale models and ML pipelines (Spark, Airflow), develop/fine-tune LLMs and AI Agents, run experimentation (A/B tests), mentor senior ICs, partner cross-functionally, and embed responsible AI practices.

Develop and productionize AI features from PoC to scalable deployments. Write production-ready Python code, implement testing and debugging, refine user stories, participate in ethical and security reviews, and collaborate with stakeholders to align AI solutions with business needs.

The Staff AI/ML Engineer will lead the development of AI systems for regulatory compliance, focusing on ML architecture, implementation, and oversight of model reliability.

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.

Design, build, and own ML/LLM systems for operations: forecasting, schedule risk detection, document extraction, and agentic systems. Handle end-to-end model lifecycle, deploy and evaluate in production, and integrate predictions into operational tools.

Lead end-to-end ML systems for identity verification, including document authenticity, face match, liveness and deepfake detection. Build GNN-based identity graphs, behavioral/device-intelligence models, vendor benchmarking/evaluation, and production enforcement. Mentor engineers and align ML architecture with Product, Compliance, Risk, and Security stakeholders.

Lead design, build, and deployment of scalable ML/AI models (including image/vision) to production. Collaborate with engineering and business teams, monitor model quality and bias, and apply data-driven solutions to predict and aggregate risk.

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.

As an ML Ops Infrastructure Engineer, you will design CI/CD pipelines for ML, maintain deployment systems, and implement monitoring while collaborating with research teams to ensure model quality and performance.

The Sr. Machine Learning Engineer will develop and deploy ML solutions for healthcare, manage data pipelines, and work with large datasets to enhance healthcare delivery.

Design and deploy production-grade machine learning models and data pipelines across image, video, text, geospatial, time-series, and structured data. Collaborate with customers, data scientists, software and DevOps engineers to define requirements, maintain mission-critical deployments, and contribute to the Chariot AI operations platform.

Lead design, training, and deployment of large-scale ML models (LLMs, diffusion, GNNs). Own end-to-end training pipelines, distributed training, evaluation, and inference architecture. Collaborate with data and infra teams, drive reproducible model development, and provide technical leadership and mentorship from research through production.

Design, build, and deploy autonomous AI agents using LLMs and agent frameworks; integrate agents into e-commerce and supply chain systems; implement multi-agent orchestration, memory and tool integrations; fine-tune models and apply RL concepts; and build safety, monitoring, and MLOps pipelines.

Design, build, deploy, and operate production-grade AI/ML and LLM systems (including RAG and agentic pipelines). Integrate AI capabilities with product workflows, run evaluation and monitoring, optimize model and system performance, and apply strong engineering practices (testing, CI/CD, versioning).

Design, develop, fine-tune, and deploy generative AI models (GANs, VAEs, transformers). Collaborate with cross-functional teams, troubleshoot model issues, document work, and communicate technical concepts to non-technical stakeholders while staying current on research.

Design, develop, and deploy machine learning and deep learning models; build end-to-end ML pipelines including preprocessing, training, evaluation, and deployment; perform feature engineering on large datasets; deploy and monitor models on cloud platforms; collaborate with cross-functional teams to integrate AI solutions and retrain models as needed.

Design, build, and deploy production-grade ML and AI systems end-to-end, combining classical ML, deep learning, and LLM/agentic integrations. Implement MLOps, monitoring, CI/CD, model serving, vector search/RAG pipelines, and scalable APIs; mentor engineers and drive engineering best practices.

Design, deploy, and optimize cloud-native ML systems on AWS, integrate foundation models (Bedrock/Anthropic), build and monitor SageMaker pipelines and data pipelines, detect drift, extend ML infra and libraries, and collaborate cross-functionally to deliver secure, cost-efficient production ML services.

Design, develop, and deploy end-to-end machine learning models using statistical and ML algorithms. Prepare and enrich data, implement scalable PySpark/Databricks solutions, and collaborate with product and engineering to create personalized customer experiences and actionable insights.

Design and productionize ML models and agentic systems for intraday financial alpha generation. Build data pipelines, experiment frameworks, and portfolio optimization methods. Collaborate with quants and engineers to move models from research to live trading environments.

Design, build, and operate large-scale ML systems across data, training, evaluation, inference, and deployment. Lead GPU-optimized training and inference pipelines, ensure reliability, performance, safety, and seamless integration into products. Own production deployments, monitoring, and cross-functional best practices to drive measurable model improvements and efficient research-to-production cycles.