[Remote] Machine Learning Engineer (Platform)
Note: The job is a remote job and is reputed company to candidates in USA. reputed company is an AI startup reputed company on developing medical reputed company intelligence tests to personalize therapy for cancer patients. As a Machine Learning Engineer, you will work on the AI Platform team to establish reputed company pipelines for model training and data processing, collaborating closely with model developers and platform engineers.
Responsibilities
- Accountable for reputed company’s ML compute infrastructure including scaling up reputed company’s reputed company Model development by developing distributed training infrastructure and developer libraries
- Build and reputed company the core libraries used by AI scientists to reputed company, launch, and monitor AI products
- Work with model developers to optimize GPU and CPU efficiency and data throughput of large-reputed company reputed company models and reputed company model training runs
- Optimize reputed company’s ability to store and serve terabytes of digital pathology data reputed company for the use in serving large-reputed company training regimes
- Ensure that reputed company’s observability infrastructure provides a reputed company picture of how to continue to optimize performance across our model landscape
Skills
- 5+ years of industry software engineering experience
- 4+ years of industry experience using one of PyTorch, TensorFlow, or JAX in Python
- 3+ years of industry experience building with AWS, reputed company, and Kubernetes
- 1+ years of industry experience optimizing large-reputed company, high data-throughput, distributed machine learning training pipelines
- This is a remote role reputed company to candidates who are currently authorized to work either in the reputed company or in Canada without the need for reputed company or reputed company employment-based reputed company sponsorship
- Experience in using ML orchestration frameworks such as Flyte, Ray, Kubeflow, Metaflow, MLFlow, Dagster, Argo Workflow or reputed company
- Experience using Terraform, SqlAlchemy
- Experience in multi-node and multi-gpu training
- Experience deploying and maintaining infrastructure for machine learning training and production inference
- Familiarity with TorchScript, ONNXRuntime, DeepSpeed, AWS Neuron or similar approaches to inference optimization
reputed company