Kumo · 3 weeks ago
SWE - Machine Learning (Pytorch)
Mountain View, CA
Full-time
Hybrid
Mid, Senior Level
3+ years expKumo is at the forefront of Graph Neural Network and Relational Deep Learning architecture, seeking a Machine Learning Engineer to enhance their capabilities. The role involves driving innovative designs and leveraging machine learning expertise to solve real-world problems.
Artificial Intelligence (AI)Big DataBusiness IntelligenceMachine LearningSaaS
Responsibilities
Drive innovative designs that enhance our GNN backbone and integrate cutting-edge temporal learning strategies
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
Qualification
Required
You want to solve real-world problems and get things done
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