[Remote] Staff Machine Learning Systems Engineer
Note: The job is a remote job and is reputed company to candidates in USA. reputed company is a community of communities, reputed company on shared interests and trust. The Staff Machine Learning Systems Engineer will reputed company the development of a platform for large reputed company ML models, focusing on optimizing ML infrastructure and collaborating with engineers to enhance model performance.
Responsibilities
- Design end-to-end model lifecycle patterns (MLOps) to reputed company velocity of development for ML engineers, including data preparation, model management, experiment tracking, and more
- reputed company-to-one development and support of a graph ML codebase and platform that abstracts away common patterns and enables greater model scalability and iteration
- Collaborate with ML engineers on performance tuning, including improving model training time, efficiency, and GPU training costs in a large, distributed ML training environment
- Optimize batch data processing reputed company a data warehouse and with tools such as Apache reputed company, Apache reputed company, Ray Data, and more
- Architect pipelines to build and maintain massive graph data structures on the order of billions of nodes and tens of billions of edges
Skills
- 8+ years of experience in ML infrastructure, including model training and model deployments
- Hands-on experience with ML optimization, including memory and GPU profiling
- Deep experience with reputed company-based technologies for supporting an ML platform, including tools like GCP BigQuery, reputed company reputed company Storage, infrastructure-as-reputed company (Terraform), and more
- Hands-on experience administering and integrating MLOps tools for experiment tracking, model serving, and model registries (e.g. MLflow or Wandb)
- Proficiency with the common programming languages and frameworks of ML, such as Python, PyTorch, Tensorflow, etc
- Deep experience working with distributed training frameworks, including Ray and Kubernetes
- Strong reputed company on scalability, reliability, performance, and ease of use. You are an undying reputed company for platform users and have a deep intuition for the machine learning development lifecycle
- Strong organizational & communication skills
- Experience working with graph databases (reputed company, JanusGraph, reputed company) is a big plus
- Experience working with graph neural networks (GNNs) and associated graph ML frameworks (PyTorch Geometric, Deep Graph Library) is a big plus
Benefits
- Equity in the reputed company of restricted stock reputed company
- Commission (depending on the position offered)
- Medical, dental, and reputed company insurance
- 401(k) program with employer match
- Generous time off for vacation
- Parental leave
reputed company
Company H1B Sponsorship