Machine Learning Engineer (AI/ML)

Overview

Design, build and operate ClickHouse's AI and machine learning products end-to-end, including the Agentic Data Stack, AI Functions, chDB, the in-Console copilot and the AI/ML partnerships that distribute them.

What you'll do

• Design and implement AI-powered features across the full stack, from backend inference services to intuitive frontend interfaces within the ClickHouse Cloud platform.
• Create robust, scalable APIs that connect ClickHouse's database capabilities with modern AI/ML inference systems and external/internal AI services.
• Implement and maintain integrations with the broader AI/ML ecosystem and standards, ensuring ClickHouse works seamlessly with popular frameworks and tools.
• Integrate models into production systems with proper monitoring, versioning, observability, and evaluation.
• Participate in the daytime oncall rotation along with the rest of the team.
• Own the ML pipeline end to end (data, training, evaluation), and fine-tune or train custom models when off-the-shelf ones fall short.
• Track the state of the art closely enough to apply what genuinely improves our products.

What you'll need

• 5+ years of software engineering experience in production environments.
• Exposure to working directly with AI/ML technologies.
• Experience integrating and deploying AI/ML models in production systems, including working with inference APIs and vector databases.
• A high level of ownership and the ability to drive features from concept to production with minimal supervision.
• Ability to thrive in collaborative environments and communicate technical concepts effectively to diverse stakeholders.
• Applied ML experience in Python (Jupyter, PyTorch, common ML libraries), hands-on with compute clusters (Ray, Slurm), and designs trustworthy experiments to drive architectural decisions.

Nice to have

• Familiarity with cloud technologies such as AWS, Azure, or GCP, particularly services related to AI/ML deployment.
• Understanding of database systems and data processing pipelines, with ClickHouse experience being a significant plus.
• Experience with embedding-based systems (retrieval, ranking, recommenders) and the representation-learning ideas underneath them.

Details

• Location: New York.
• ClickHouse is remote-friendly and operates in over 25 countries; a flexible work environment is offered.