Affinity.co

The CRM for Deal Makers. Relationship Intelligence, Reimagined.

Senior AI Engineer

AI EngineerMachine Learning EngineerFull TimeRemoteTeam 201-500Since 2014H1B No SponsorCompany SiteLinkedIn

Location

United States

Posted

2 days ago

Salary

$160K - $235K / year

PythonMachine LearningLLMRAGVector SearchHybrid RetrievalRerankingContextual CompressionPrompt EngineeringText2sqlEntity ExtractionVector DatabasesGraph DatabasesEmbedding ModelsContext Window OptimizationMetadata RetrievalModel EvaluationMlopsCi/cdPipeline Automation

Job Description

This description is a summary of our understanding of the job description. Click on 'Apply' button to find out more.

Role Description

Affinity stitches together billions of data points from massive datasets to create a powerful, accurate representation of the world's professional relationship graph. Based on this data, we offer our users the insights and visibility they need to nurture and tap into the opportunities in their team's network. This role is part of the AI Platform team, which owns the AI services that power Affinity's industry-leading relationship intelligence platform. We extract and retrieve information from billions of structured and unstructured data points to deliver actionable insights to customers.

As a Senior AI Engineer, you will collaborate with machine learning engineers, data engineers, software engineers, and product managers to shape the future of private capital's leading CRM platform. You will design and build LLM-powered AI systems that efficiently uncover insights from compelling business interaction data – an exciting and unique opportunity within the industry.

  • Build RAG systems:
    • Architect, prototype, and deploy RAG pipelines, combining vector search, hybrid retrieval, reranking and contextual compression techniques.
  • Build LLM powered agent systems:
    • Contribute to design and orchestration of multi-agent LLM systems using community frameworks and custom orchestration layers.
  • Solve complex problems:
    • Work on a variety of information extraction, information storage and information retrieval problems for both structured and unstructured data.
  • Collaborate cross-functionally:
    • Partner with cross-functional (product, infra, data engineering, and software engineering) to build robust, high-scale systems that underlie all of our data processing and ML Operations.

Qualifications

  • 5+ years of experience in software engineering and/or Machine Learning experience in applying machine learning in production.
  • Hands on experience with LLM applications in production including prompt engineering and utilizing frameworks for online and offline evaluation.
  • Experience with LLM assisted search, such as query understanding and augmentation, text2sql, and entity extraction.
  • Experience with vector or graph databases.
  • Experience with document chunking, embedding models, and context window optimization.
  • Familiarity with metadata-based retrieval and re-ranking strategies.
  • Hands on experiences with model evaluation metrics (e.g. perplexity, hallucination rate, factual consistency).
  • Familiarity with data security, versioning, and MLOps principles.

Requirements

  • Experience with enterprise AI applications with strict compliance, audit, or legal requirements.
  • Experience with dataset engineering, including data curation, augmentation, and synthesis, to assist ML model improvements.
  • Experience with multi-modal search.
  • Experience with graph based recommendation systems, such as graph NN.
  • Experience with developing AI applications powered by agent-based systems.
  • Experience with packaging, CI/CD and pipeline automation.

Benefits

  • Health Benefits: We cover your medical, dental, and vision insurance premiums with comprehensive PPO, HDHP and HMO options (in CA), and offer flexible personal & sick days to support your well-being.
  • Retirement Planning: We offer a 401(k) plan to help you plan for your future.
  • Learning & Development: We provide an annual education budget and a comprehensive L&D program.
  • Wellness Support: We reimburse monthly for things like home internet, meals, and wellness memberships/equipment to support your overall health and happiness.
  • Team Connection: Virtual team-building activities and socials to keep our team connected, because building strong relationships is key to success.

Company Description

With more than 3,000 customers worldwide and backed by some of Silicon Valley's best firms, Affinity has raised $120M to empower dealmakers to find, manage, and close more deals. Our Relationship Intelligence platform uses the wealth of data exhaust from trillions of interactions between Investment Bankers, Venture Capitalists, Consultants, and other strategic dealmakers to deliver automated relationship insights that drive over 450,000 deals every month. We are proud to have received Inc. and Fortune Best Workplaces awards as well as to be Great Places to Work certified for the last 5 years running. Join us on our mission to make it possible for anyone to cultivate and fully harness their network to succeed.

Job Requirements

  • 5+ years of experience in software engineering and/or Machine Learning experience in applying machine learning in production.
  • Hands on experience with LLM applications in production including prompt engineering and utilizing frameworks for online and offline evaluation.
  • Experience with LLM assisted search, such as query understanding and augmentation, text2sql, and entity extraction.
  • Experience with vector or graph databases.
  • Experience with document chunking, embedding models, and context window optimization.
  • Familiarity with metadata-based retrieval and re-ranking strategies.
  • Hands on experiences with model evaluation metrics (e.g. perplexity, hallucination rate, factual consistency).
  • Familiarity with data security, versioning, and MLOps principles.
  • Experience with enterprise AI applications with strict compliance, audit, or legal requirements.
  • Experience with dataset engineering, including data curation, augmentation, and synthesis, to assist ML model improvements.
  • Experience with multi-modal search.
  • Experience with graph based recommendation systems, such as graph NN.
  • Experience with developing AI applications powered by agent-based systems.
  • Experience with packaging, CI/CD and pipeline automation.

Benefits

  • Health Benefits: We cover your medical, dental, and vision insurance premiums with comprehensive PPO, HDHP and HMO options (in CA), and offer flexible personal & sick days to support your well-being.
  • Retirement Planning: We offer a 401(k) plan to help you plan for your future.
  • Learning & Development: We provide an annual education budget and a comprehensive L&D program.
  • Wellness Support: We reimburse monthly for things like home internet, meals, and wellness memberships/equipment to support your overall health and happiness.
  • Team Connection: Virtual team-building activities and socials to keep our team connected, because building strong relationships is key to success.

Related Job Pages

More AI Engineer Jobs

AI Consultant

Southwest Airlines

Southwest Airlines is an Equal Opportunity Employer. Please print/save this job description because it won't be available after you apply. Job Posting End Date: 03/13/2026

AI Engineer2 days ago
Full TimeRemoteTeam 10,001

The AI Consultant supports business stakeholders and delivery teams in shaping and delivering AI, ML, Generative AI, and Agentic AI solutions that address meaningful business needs. They will lead the end-to-end lifecycle of AI products, from opportunity identification through delivery, while collaborating with various groups.

United States
$125K - $139K / year

Sr AI/ML Engineer Remote - Eastern Time Zone

Optum

Optum, part of the UnitedHealth Group family of businesses, is a global organization that delivers care, aided by technology to help millions of people live healthier lives. The work you do with our team will directly improve health outcomes by connecting people with the care, pharmacy benefits, data and resources they need to feel their best. Here, you will find a culture guided by inclusion, talented peers, comprehensive benefits and career development opportunities. Come make an impact on the communities we serve as you help us advance health optimization on a global scale. Join us to start Caring. Connecting. Growing together. At Optum, we support your well-being with an understanding team, extensive benefits and rewarding opportunities. By joining us, you’ll have the resources to drive system transformation while we help you take care of your future. We recognize the power of connection to drive change, improve efficiency and make a difference in health care. Join a team where your skills and ideas can make an impact and where collaboration is key to creating technology that produces healthier outcomes.

AI Engineer2 days ago
Full TimeRemoteTeam 160,000Since 2011

Design, develop, and deploy generative AI solutions (LLMs, RAG, Agentic AI) for Pharmacy; build and productionize models using Python and deep learning frameworks; implement CI/CD, containerization, cloud deployments, testing, and performance optimization while collaborating with Pharmacy Operations and Capability teams.

Agentic AiAzureCI/CDContainerizationGitGithub ActionsKerasLangchainLanggraphLlmsPythonPyTorchRagTensorFlow
Minnesota
$91.7K - $163.7K / year

Sr AI/ML Engineer - Remote Nationwide or Hybrid in NJ, MN, DC

Optum

Optum, part of the UnitedHealth Group family of businesses, is a global organization that delivers care, aided by technology to help millions of people live healthier lives. The work you do with our team will directly improve health outcomes by connecting people with the care, pharmacy benefits, data and resources they need to feel their best. Here, you will find a culture guided by inclusion, talented peers, comprehensive benefits and career development opportunities. Come make an impact on the communities we serve as you help us advance health optimization on a global scale. Join us to start Caring. Connecting. Growing together. At Optum, we support your well-being with an understanding team, extensive benefits and rewarding opportunities. By joining us, you’ll have the resources to drive system transformation while we help you take care of your future. We recognize the power of connection to drive change, improve efficiency and make a difference in health care. Join a team where your skills and ideas can make an impact and where collaboration is key to creating technology that produces healthier outcomes.

AI Engineer2 days ago
Full TimeRemoteTeam 160,000Since 2011

Lead design, development, deployment, and maintenance of AI-powered solutions (LLMs, Generative AI, traditional ML) for provider-facing products. Build chatbots and conversational NLP, integrate AI into cloud (preferably AWS), ensure responsible AI practices and compliance, provide technical leadership, mentor engineers, and collaborate with product and architecture teams.

Api DevelopmentAWSChatbotsCloudConversational AiGenerative AiLlmsMachine LearningNlpWeb Development
Minnesota
$91.7K - $163.7K / year

Principal AI/ML Engineer - Remote

Optum

Optum, part of the UnitedHealth Group family of businesses, is a global organization that delivers care, aided by technology to help millions of people live healthier lives. The work you do with our team will directly improve health outcomes by connecting people with the care, pharmacy benefits, data and resources they need to feel their best. Here, you will find a culture guided by inclusion, talented peers, comprehensive benefits and career development opportunities. Come make an impact on the communities we serve as you help us advance health optimization on a global scale. Join us to start Caring. Connecting. Growing together. At Optum, we support your well-being with an understanding team, extensive benefits and rewarding opportunities. By joining us, you’ll have the resources to drive system transformation while we help you take care of your future. We recognize the power of connection to drive change, improve efficiency and make a difference in health care. Join a team where your skills and ideas can make an impact and where collaboration is key to creating technology that produces healthier outcomes.

AI Engineer2 days ago
Full TimeRemoteTeam 160,000Since 2011

Lead end-to-end AI/ML projects: research, prototype, optimize, and productionize ML and LLM solutions for healthcare. Mentor and recruit teams, run experiments and A/B tests, present results to leadership, and pursue AI research and IP.

Deep LearningDpoGansGenerative AiGnnsLarge Language ModelsMultimodalNlpPpoPythonPyTorchSsmsTensorFlowTransformers
Texas + 4 moreAll locations: Texas, Georgia, Virginia, Washington, California
$134.6K - $230.8K / year