Top Machine Learning Jobs in Buffalo, NY

The AI Engineer will design and build production systems, focusing on LLM integration, system reliability, evaluation frameworks, and high-quality product features while maintaining strong engineering practices.

As a Principal Product Manager, you'll lead product strategy, collaborate across teams, utilize data for decision-making, and partner with top executives to drive product success.

As a Software Engineering Intern, you will contribute to building features for user-facing products, APIs, and scalable services for machine learning. Responsibilities include crafting code, designing data pipelines, implementing tools, and enhancing technical documentation.

The role involves building predictive models for blockchain performance and user behavior, implementing solutions, and owning production prediction pipelines.

Lead end-to-end delivery of LOIS by owning the master program plan, coordinating cross-functional teams (engineering, product, design, ML, data, SRE, operations), resolving blockers and dependencies, establishing lightweight delivery infrastructure and tooling, partnering on release planning, and maintaining transparent status, risks, and accountability to deliver complex features to customers.

This role involves leading a perception team in developing algorithms for autonomous driving systems, supervising team members, and conducting research to enhance perception capabilities.

Design, optimize, and scale ClickUp's search infrastructure using OpenSearch/ElasticSearch. Improve relevance, speed, and real-time indexing; build vector/semantic search capabilities; create measurement frameworks; integrate search with AI and product features; troubleshoot at scale; and mentor junior engineers to maintain high search quality under heavy load.

Design, scale, and optimize ClickUp's search infrastructure using OpenSearch/Elasticsearch. Improve relevance, speed, and real-time indexing, build vector search capabilities, create evaluation frameworks, and collaborate with AI/backend/product teams to integrate and troubleshoot search at scale.

Manage full-cycle recruiting for technical and consulting roles with emphasis on AI talent. Partner with leaders to build sourcing strategies, pipelines, and candidate evaluation for LLMs, prompt engineering, ML, and AI governance roles. Maintain ATS data, report metrics, document recruiting best practices, and deliver strong candidate and hiring manager experiences.

The Data & Analytics team at Surescripts transforms healthcare data into actionable insights, collaborating with various stakeholders to support critical decisions across the healthcare ecosystem.

Lead Tekmetric's API product strategy and roadmap for the SaaS platform. Own API lifecycle (contracts, versioning, documentation), drive developer experience and adoption, identify integration opportunities, lead generative AI initiatives, perform user research and analytics-driven prioritization, collaborate across design/engineering/sales, train internal teams, and measure business impact.

One-year, full-time fellowship supporting design, delivery, and evaluation of interventions for smallholder farmers. Duties include research and analysis, pilot design, field implementation, M&E, and cross-team collaboration with heavy mentorship and rotating in-country placements.

Lead development of Vanta's business analytics function: build scalable product data assets, define measurement frameworks, use AI-assisted analytics, partner with cross-functional teams, and deliver executive-level recommendations to drive product and business decisions.

Lead strategy, build and scale a global partnerships ecosystem (ERP, fintech, banks, ISVs, VARs, SIs) to drive partner-sourced pipeline, revenue, and platform expansion. Own partner negotiations, commercial models, cross-functional integration, and team development while leveraging AI/data insights and ensuring compliance.

Lead the vision and execution of data product roadmap; translate business needs into epics and acceptance criteria; partner with data engineering and analytics; run agile ceremonies and manage backlog; champion data-as-a-product; scope and pilot AI/ML capabilities; present roadmaps and business cases to leadership.

Full-cycle territory sales role selling Greenway Health's EHR, RCM, and AI-driven healthcare solutions. Responsibilities include prospecting, lead generation, demos, RFP management, proposal configuration, negotiation, closing, maintaining Salesforce, meeting quota, and providing market feedback. Role requires heavy travel (50–75%) and collaboration with internal solutions consultants to deliver value-based technology and revenue cycle capabilities to ambulatory healthcare clients.

Design, build, and operate core AI capabilities and reusable model integrations for an enterprise platform. Integrate foundation models into production, optimize performance and reliability, prototype new techniques, and support scalable backend services and distributed AI systems in collaboration with cross-functional teams.

Own end-to-end product development for AI/ML revenue cycle solutions: define vision and roadmap, write specs and user stories, prioritize backlog, work closely with data scientists and engineers, drive agile delivery, ensure quality, and measure product performance with stakeholders.

Work on-site with portfolio companies to design, prototype, and deliver full-stack digital and AI solutions. Lead discovery, deployment, and post-launch iteration, ensure security/compliance, resolve technical issues, mentor teams, and drive enterprise integrations and adoption. Frequent domestic travel and stakeholder management required.

Sell Litify's AI-driven legal operating platform to law firms and legal departments. Manage full upsell motions, consult on AI adoption, deliver high-level demos to C-suite, meet sales targets, and articulate ROI of autonomous agents and AI integrations.

Lead and mature Omada's enterprise privacy program, manage HIPAA compliance and incident response, embed privacy-by-design across products, conduct privacy risk assessments and vendor due diligence, advise on consumer privacy and AI governance, support interoperability/HIE participation, and deliver privacy training and guidance across the business.

The Staff Machine Learning Engineer will lead ML systems development for personalization and ranking, design architectures, and mentor team members, ensuring reliable operations at scale.

The Machine Learning Engineer will work on unstructured data classification and clustering, utilizing their expertise in AI systems and various programming languages.

The Senior Machine Learning Engineer will utilize deep experience and insights to enhance communication and digital interactions through advanced machine learning tools and algorithms, focusing on unstructured data.

The Staff Data Scientist will lead the development and deployment of ML and GenAI systems to detect billing errors and fraud in healthcare payments, collaborating with cross-functional teams.