Staff Machine Learning Engineer, ML Platform

7+ years of experience in ML infrastructure, model training, and model deployments Hands-on experience with ML optimization (memory and GPU profiling) Deep experience with cloud-based technologies for ML platforms (GCP BigQuery, Google Cloud Storage, Terraform) Hands-on experience administering and integrating MLOps tools (MLflow, Wandb) Proficiency with Python, PyTorch, Tensorflow Deep experience with distributed training frameworks (Ray, Kubernetes) Experience working with graph databases (Neo4j, JanusGraph, TigerGraph) is a plus Experience working with graph neural networks (GNNs) and associated graph ML frameworks (PyTorch Geometric, Deep Graph Library) is a plus

Design end-to-end model lifecycle patterns (MLOps) Develop and support a graph ML codebase and platform Collaborate on performance tuning for ML models Optimize batch data processing Architect pipelines for massive graph data structures