Software Engineer - GNN Expert

Kumo in Mountain View, CA

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. Get ready to make a significant impact in a dynamic and exciting environment!
We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.
    • 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
    • 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
    • Stock
    • Competitive Salaries
    • Medical, dental insurance
Apply