Software Engineer (ML Infra)

Deep experience optimizing ML systems for GPUs and CPUs. Knows JAX or similar tools. Works with Data Scientists to build complex model training and data processing workflows. Understands fine-tuning transformer models. Dreams about optimizing BigQuery-based ETL pipelines. Slices and dices data with tools such as Pandas, SQL, and dplyr. 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.

Collaborate with Data Scientists to design, develop, and implement model training and evaluation workflows. Optimize machine learning models built with JAX and TensorFlow. Work on all parts of the data processing stack. Manage variations of models and configurations supporting custom research. Build tools to empower data analysts to derive insights from model outputs. Focus on building effective tools for data scientists and data analysts.