Senior ML & AI Engineer

About the role

PitchBook’s Machine Learning Engineer is responsible for using machine learning, statistical modeling, and natural language processing (NLP) to collect a high-volume of data for the PitchBook Platform and surface insights for the capital markets. This role will also serve as a resource and mentor for junior members of the team. A strong motivation to succeed is critical and everyone has the opportunity to shape the long-term direction of our team.

About the project

PitchBook — a platform for investment professionals. Our software provides access to data and the analytical tools to get answers fast and discover promising opportunities. Uncovers actionable insights and trends hidden within the financial data of more than three million companies. Users all over the world include large corporations, start-ups, venture capital and private equity firms, investment banks, and many others. Features of PitchBook: Advanced search / Discovery & insights / Company profiles / Workflow & efficiency / Financials and many more.

Team: 1 EM, 6 ML Engineers

Work Environment: The role offers a flexible work schedule, allowing you to adapt your working hours with the requirement to attend all team meetings. The team follows a Scrum-based Agile methodology.

Tech Stack & Focus Areas: Work with generative AI and transformer models, integrating frameworks like LangChain, LangGraph, and LangSmith to automate workflows and build intelligent systems. Use PyTorch, Scikit-Learn, and Spacy to build NLP pipelines. Implement vector databases for high-dimensional data embeddings and develop hybrid models to extract insights from diverse data sources.

As a qualified expert You will

• Work with GenAI & LLMs: Design, develop, and fine-tune generative AI models using transformer architectures. Implement advanced techniques to create practical solutions for real-world challenges.
• Use Modern Toolkits: Leverage frameworks like LangChain, LangGraph, and LangSmith to build automated pipelines and intelligent agents. Integrate these tools with vector databases for efficient similarity search and real-time analysis.
• Extract Data Insights: Process and analyze large volumes of unstructured data using embeddings and statistical modeling to identify meaningful insights.
• Build Scalable Systems: Collaborate with cross-functional teams to create production-grade ML systems that handle real-time, high-volume data processing.
• Mentor & Lead: Act as a mentor for junior team members and stay updated on trends in Agentic AI, transformer models, and generative systems.
• Collaborate & Communicate: Work with Product and Engineering teams to improve data management and support innovation. Present complex analyses clearly and effectively.
• Tech Stack & Focus Areas: Work with generative AI and transformer models, integrating frameworks like LangChain, LangGraph, and LangSmith to automate workflows and build intelligent systems. Use PyTorch, Scikit-Learn, and Spacy to build NLP pipelines. Implement vector databases for high-dimensional data embeddings and develop hybrid models to extract insights from diverse data sources.

We’re looking for you if you have

• Educational Background: You hold a Bachelor’s or Master’s degree in Computer Science, Machine Learning, Statistics, or a related field.
• Experience: You bring 4+ years of production experience in the ML field, with a track record in developing transformer-based models and deploying them in scalable systems.
• Technical Expertise:
  • Proficient in Python and machine and deep learning frameworks like TensorFlow, PyTorch, ScikitLearn, Spacy or other.
  • Experienced in natural language processing and modern LLM based techniques.
  • Experience with full ML lifecycle, data preprocessing, training, evaluation stages and monitoring techniques.
  • Skilled with SQL and data extraction methods.
• Innovative Thinker: You thrive on solving complex problems and are always exploring new trends in Generative AI.
• Collaborator & Communicator: You’re an excellent communicator able to collaborate and express your thoughts in a professional manner to team members, other engineering teams and stakeholders.
• Domain Knowledge: Exposure to financial services, investment banking, venture capital, or private equity is nice to have.

What’s in it for You

Reveal great tech solutions

Join the team of experts who create custom, cutting-edge tech solutions for world-renowned businesses, fueling client growth. Unleash your potential, tackle new challenges, and be part of a team that values your skills and contributions. Focus on long-term impact and building tailored, long-lasting partnerships with our clients.

Experience an agile and flexible working environment

Enjoy the freedom of fully remote work with a flexible working schedule. Empower yourself with a stable workload and a stable income, supported by provided laptops and licensed software. We focus on lasting cooperation and unite result-oriented individuals who stand on a high-performance approach to work.

Embrace the opportunity for personal and professional growth

Benefit from performance and merit reviews, elevate your skills with personal development plans, and individual learnings through the corporate library, public speaking support, and more.

Be among like-minded people

Work with a team of one mind who cares about what they do and how they do. Collaborate with top-notch experts who are always ready to help and support you through any challenges. Join company-wide tech and cultural events, and contribute to meaningful CSR initiatives that resonate with your values. Feel supported by your HR, and take advantage of our referral bonus program.