Blue Rose Research is looking for a software engineer with extensive machine learning infrastructure and operations experience to design, develop, and implement sophisticated model training and evaluation workflows, optimize machine learning models, and manage variations of models and configurations to support custom research on tight deadlines.
Requirements
- Has deep experience optimizing ML systems for GPUs and CPUs.
- Knows JAX or similar tools inside and out.
- Regularly works with Data Scientists to build complex model training and data processing workflows.
- Understands the ins and outs of fine-tuning transformer models.
- Dreams about optimizing BigQuery-based ETL pipelines.
- Slices and dices data with a variety of tools such as Pandas, SQL, and dplyr.
- Has a solid DevOps toolkit including Docker, Kubernetes, Terraform, etc.
Responsibilities
- Collaborate with Data Scientists to design, develop, and implement highly sophisticated model training and evaluation workflows.
- Optimize machine learning models built with JAX and TensorFlow.
- Work on all parts of the data processing stack, from the initial data ingestion and cleaning, through model training, model evaluation, and model inference.
- Manage many variations of different models and model configurations supporting custom research on tight deadlines.
- Build tools to empower data analysts to derive insights from model outputs.
- Stay focused on building effective tools for data scientists and data analysts without getting distracted by shiny engineering problems.
Other
- Candidates must be authorized to work lawfully in the United States.
- Has strong oral and written communication skills, especially in a remote environment.
- Is a kind person and a team player who contributes to a warm working environment.
