Software Engineer

Summary

Join Blue Rose Research, a leading progressive organization, as a Software Engineer specializing in machine learning infrastructure and operations. Collaborate with data scientists to design, develop, and implement sophisticated model training and evaluation workflows. Optimize machine learning models built with JAX and TensorFlow, manage model variations, and build tools for data analysts. The role requires expertise in optimizing ML systems for GPUs and CPUs, working with data scientists on complex workflows, and understanding transformer model fine-tuning. This remote position offers a competitive salary, medical, dental, and health benefits, and a supportive work environment. Most work happens on East Coast time, with an office in NYC and regular meetups in DC.

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
• Is familiar with a variety of machine learning and statistical techniques
• Has a solid DevOps toolkit including Docker, Kubernetes, Terraform, etc
• 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
• Candidates must be authorized to work lawfully in the United States

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

Benefits

• Competitive salary
• Medical, dental, and health benefits