Top Machine Learning Jobs in Madison, WI

Design, optimize, and operate production LLM inference and training pipelines. Improve latency, throughput, and GPU utilization using techniques like batching, quantization, FlashAttention, and kernel-level profiling. Deploy and autoscale multiple models on shared GKE GPU clusters, consult with clients on performance and cost requirements, and carry prototypes to robust, scalable production services.

Design, develop, and deploy end-to-end AI/ML applications (RAG, chatbots, agentic workflows). Build scalable data pipelines, apply graph analytics, entity resolution, and NLP for risk scoring, anomaly detection, and investigative prioritization. Develop models, REST APIs, and deploy/manage solutions in AWS to support federal mission use cases.

Build and re-architect Shipt's LLM-native search stack including retrieval and ranking systems. Design multi-task and multi-objective models, integrate personalization signals, develop hybrid dense/sparse retrieval, and apply generative retrieval and query expansion. Analyze search data, report findings to leadership, and partner with senior scientists and ML engineers to improve relevance, revenue trade-offs, and operational metrics.

Develop and scale reinforcement-learning systems that close the loop between automated labs and reward models. Build synthetic data engines and inference infrastructure for large-scale training, maintain evaluations of learned policies using proprietary experimental datasets, and address real-world challenges like experimental noise and expensive queries.

As a Senior Machine Learning Engineer, you'll design and develop infrastructure and models for AI-powered applications while collaborating with teams to enhance product functionality.

As a Senior Staff Machine Learning Engineer, you will lead the development and deployment of machine learning models, ensuring they drive business growth and meet performance requirements. You will collaborate with teams, manage data pipelines, and implement state-of-the-art solutions while continuously improving model performance.

The role involves developing and deploying machine learning models for forecasting within the Camus platform, requiring collaboration and technical depth in ML.

Lead design, training, fine-tuning, evaluation, and production deployment of TTS/voice-generation models. Translate research into reliable, low-latency production systems, define ML best practices, and collaborate across research, engineering, and product to drive execution.

Design and develop AI capabilities and scalable microservices. Collaborate to integrate AI models and ensure performance in cloud environments.

As an MLOps/ML Platform Engineer, you will build and manage ML systems, optimize workloads, and ensure production model reliability while collaborating across teams.

Seeking a Senior Machine Learning Engineer to design and implement ML systems, optimize deployment, and collaborate with data teams, focusing on scalability and performance.

Manage AI/ML projects from Proof of Concept to Minimum Viable Product using Agile methodologies while ensuring stakeholder communication and project alignment.

The R&D Engineer will build and refine software solutions for defense use cases, prototype systems, and convert stakeholder feedback into technical requirements. Collaboration with engineering teams and customer-oriented engagements are key responsibilities.

Build and deploy production ML models and end-to-end pipelines to detect suspicious behavior; design evaluation, monitoring, and drift detection; integrate intelligence into product workflows; improve data quality and accessibility; and contribute to operational rigor, runbooks, and safe rollout practices in a security-sensitive environment.

As a Software Engineer in ML Infrastructure, you will develop scalable data pipelines, enhance data discovery, and collaborate with teams on data organization and ML features.

The Senior AI/ML Engineer will develop AI systems for enterprise decision-making, focusing on LLMs and real-time data processing. Responsibilities include building detection systems, insight engines, evaluation frameworks, and collaborating with engineers to ensure robust AI product engineering.

Design and implement ML infrastructure for scalable deployment, collaborate across teams, optimize ML systems, and enhance tooling.

As a Python and Kubernetes Software Engineer, you will develop data analytics solutions using open source tools, collaborating with a distributed team.

Develop AI-powered solutions for personalized content and marketing. Collaborate with teams to optimize ML models and pipelines at scale, mentoring engineers and driving strategic growth initiatives.

The Senior Databricks AI/ML Engineer will design and deploy AI/ML solutions in the insurance sector, focusing on Databricks architecture, data pipelines, and MLOps workflows, while collaborating with various teams to operationalize models and optimize processes.

The Solutions Engineer will drive the adoption of AI-powered solutions in the electric utility sector, providing technical consulting and customer enablement. Responsibilities include gathering requirements, architecting solutions, leading RFI/RFP processes, and delivering training. Extensive travel is required for client engagement.

Design, implement, and scale web and data pipeline solutions focused on generative AI and LLMs. Improve AI/ML infrastructure for model development, training, deployment, and prompt engineering. Partner with product and stakeholders to prioritize applied AI use cases, ensure PHI security and regulatory compliance, mentor junior engineers, and contribute APIs and integrations to embed generative AI into healthcare applications.

The VP of Research, Machine Learning will set research direction for AI systems, focusing on context representation, reasoning, and evaluation frameworks while ensuring product safety and alignment.

Responsible for building and maintaining end-to-end ML pipelines and deploying production-grade ML systems. Tasks include model fine-tuning, evaluation, and collaboration with engineering teams.

As a Staff Machine Learning Engineer - AI Tech Lead, you will lead the design of AI systems for security operations, focusing on scalable architectures, AI safety, and mentoring. Responsibilities include overseeing technical evaluations, production workflows, LLM fine-tuning, and collaboration with teams to enhance AI capabilities.