Kumo · 3 weeks ago
SWE - Machine Learning (Pytorch)
Mountain View, CA
Full-time
Onsite
Mid, Senior Level
3+ years expKumo is seeking a Machine Learning Engineer specializing in Graph Neural Networks and Deep Learning. The role involves driving innovative designs and integrating learning strategies to enhance their GNN capabilities while addressing real-world problems.
Artificial Intelligence (AI)Big DataBusiness IntelligenceMachine LearningSaaS
Responsibilities
Join us at the heart of the Kumo Graph Neural Network and Relational Deep Learning architecture, where you'll play a pivotal role in shaping and expanding its capabilities
You'll drive innovative designs that enhance our GNN backbone and integrate cutting-edge temporal learning strategies
Your work will bridge the gap between foundational models and GNNs, tackling diverse use cases like recommendation systems, customer retention, forecasting, and fraud detection
Leverage your expertise in machine learning and AI to solve critical challenges, creating scalable and adaptable solutions that push the boundaries of what’s possible
Qualification
Required
Ph.D. in Computer Science or Machine Learning or related technical discipline, or related practical experience
3+ years experience in designing ML algorithm solutions
2+ graph-based machine learning papers at top venues such as NeurIPS, ICLR, ICML, etc
Well experienced in Python and its deep learning framework ecosystem, such as PyTorch
Preferred
Passionate about OSS, including various contributions
Experience developing large-scale systems
Ability to diagnose technical problems, debug code, and automate routine tasks
Analytical approach coupled with solid communication skills and a sense of ownership
Benefits
Stock
Competitive Salaries
Medical, dental insurance
Company
Kumo
Kumo is an AI model company that provides a platform for anyone to train and run state-of-the-art AI models on their relational data.
51-200 employees
H1B Sponsorship
Kumo has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (8)
2024 (3)
2023 (3)
2022 (2)
Funding
Current Stage
Growth StageTotal Funding
$36.5MKey Investors
Sequoia Capital
2022-09-27Series B· $18M
2022-04-07Series A· $18.5M
Leadership Team
Dong Wang
Founding Engineer, Engineering Leader - Product Engineering
Recent News
2025-12-02
Company data provided by crunchbase