Software Engineer - Infrastructure
Full-time
Kumo.AI
http://kumo.ai
Mountain View, CA
Founded:
2021
Software Development
Employees:
58
Posted Date:
20 August 2025
Company Details
We are among the TOP 50 AI companies globally according to Forbes, dedicated to Democratizing AI on the Modern Data Stack. The team behind PyG (PyG.org) is working on a turn-key solution for AI over large scale data warehouses. We believe the future of ML is a seamless integration between modern cloud data warehouses and AI algorithms. Our ML infrastructure massively simplifies the training and deployment of ML models on complex data. With over 40,000 monthly downloads and nearly 13,000 Github stars, PyG is the ultimate platform for training and development of Graph Neural Network (GNN) architectures. GNNs -- one of the hottest areas of machine learning now -- are a class of deep learning models that generalize Transformer and CNN architectures and enable us to apply the power of deep learning to complex data. GNNs are unique in a sense that they can be applied to data of different shapes and modalities.
Specialities
Highlights
INDUSTRIES
Software Development
SENIORITY LEVEL
Mid-Senior level
experience, years
5
Software position
yes
Job function
Engineering and Information Technology
Core Tech Stack
Nice to have
Experience with ML frameworks like PyTorch or TensorFlow, contributions to open-source projects
RESPONSIBILITIES
Important info
BS required, MS/PhD preferred, strong foundation in computer science
Kumo.AI
http://kumo.ai
Mountain View, CA
Founded:
2021
Software Development
Employees:
58