Job Title: SWE - Machine Learning (Pytorch)

Company Name: Kumo AI

Job Details: RemoteFull,Time

Job Url: https://hiring.cafe/viewjob/b2bx383h9jo8s6yc

Job Description: Posted 2mo agoSWE - Machine Learning (Pytorch)@ Kumo AIView All JobsWebsiteMountain View or London or United StatesRemoteFull TimeResponsibilities:Design algorithms, Develop models, Collaborate across teamsRequirements Summary:PhD in CS/ML or related experience; 3+ years designing ML algorithms; 2+ graph ML papers; strong Python and PyTorch; OSS contributions; large-scale systems experience.Technical Tools Mentioned:Python, PyTorch
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!Your Foundation: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 solutions2+ graph-based machine learning papers at top venues such as NeurIPS, ICLR, ICML, etcWell experienced in Python and its deep learning framework ecosystem, such as PyTorchYour Extra Special Sauce:Passionate about OSS, including various contributionsExperience developing large-scale systemsAbility to diagnose technical problems, debug code, and automate routine tasksAnalytical approach coupled with solid communication skills and a sense of ownershipBenefitsStockCompetitive SalariesMedical, dental insuranceWe 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.