Ph.D. in Computer Science, Machine Learning, or a related field, or equivalent practical experience., 3+ years of experience in designing machine learning algorithm solutions., Published at least 2 graph-based machine learning papers in top venues like NeurIPS, ICLR, or ICML., Proficient in Python and deep learning frameworks such as PyTorch..
Key responsibilities:
Drive innovative designs to enhance the GNN backbone and integrate temporal learning strategies.
Bridge the gap between foundational models and GNNs for various use cases like recommendation systems and fraud detection.
Leverage machine learning and AI expertise to solve critical challenges and create scalable solutions.
Collaborate in a dynamic environment to tackle real-world problems and implement effective solutions.
Report This Job
Help us maintain the quality of our job listings. If you find any issues with this job post, please let us know.
Select the reason you're reporting this job:
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.
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 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
Your Extra Special Sauce:
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
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.
Required profile
Experience
Spoken language(s):
English
Check out the description to know which languages are mandatory.