Machine Learning Engineer - Data & Evaluation

Summary

Join Hume AI, an AI research lab and startup, as a Software Engineer to contribute to building empathy and goal-alignment into AI. Partner with researchers to collect and process data for training multimodal LLMs. Create and implement evaluations, manage training data pipelines, and integrate open source packages and frameworks to accelerate training. This role combines data platform development with research engineering and is located in New York City or remote. Hume AI is committed to ensuring AI is guided by human values and offers a competitive salary.

Requirements

• Have expertise in the Python ecosystem and popular ML libraries and tools (e.g. PyTorch, JAX, TensorFlow)
• Are familiar with tools such as Elasticsearch, Ray, Dask, Flink, Spark
• Enjoy translating research questions into concrete experiments backed by data and measurement
• Have experience with web development for interactive tools that accelerate research
• Are comfortable designing data schemas and manipulating large data sources in BigQuery, Redshift, etc
• Excellent communication and collaboration skills

Responsibilities

• Partner closely with researchers to collect and process essential data for training multimodal LLMs
• Create and implement evaluations
• Manage training data pipelines
• Integrate open source packages and frameworks to accelerate training

Benefits

$170,000 — $250,000 USD