about 4 hours ago
Remote, United States
Staff+
H1B Sponsor
Base Salary
$230k - $322k/yr
Responsibilities
- Design end-to-end model lifecycle patterns to enhance ML development velocity.
- Develop and support a graph ML codebase for improved model scalability.
- Collaborate with ML engineers on performance tuning and efficiency improvements.
- Optimize batch data processing using tools like Apache Beam and Apache Spark.
- Architect pipelines for maintaining large graph data structures.
Requirements
- 8+ 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