about 4 hours ago
Remote, United States
Senior / Mid Level
H1B Sponsor
Base Salary
$217k - $303k/yr
Responsibilities
- Design end-to-end model lifecycle patterns (MLOps) to enhance ML development velocity.
- Develop and support a graph ML codebase and platform for scalability.
- Collaborate with ML engineers to optimize model training performance.
- Optimize batch data processing using tools like Apache Beam and Apache Spark.
- Architect pipelines for managing large graph data structures.
Requirements
- 5+ years of experience in ML infrastructure, including model training and deployments.
- Hands-on experience with ML optimization, including memory and GPU profiling.
- Deep experience with cloud technologies like GCP BigQuery and Google Cloud Storage.
- Experience with MLOps tools for experiment tracking and model serving.
- Proficiency in programming languages and frameworks such as Python, PyTorch, and TensorFlow.
- Experience with distributed training frameworks like Ray and Kubernetes.
- Strong organizational and communication skills.
- Experience with graph databases and graph neural networks is a plus.
Benefits
- Medical, dental, and vision insurance.
- 401(k) program with employer match.
- Generous time off for vacation.
- Parental leave.
Tech Stack
Apache BeamApache SparkGoogle CloudKubernetesMLflowNeo4jPythonPyTorchTensorFlowTerraform
Categories
AI & MLData Engineering