[Remote] Machine Learning Engineer

Note: The job is a remote job and is open to candidates in USA. Higharc is seeking a hands-on Machine Learning Engineer to build and ship models that understand homes and support its generative floor plan product. The role owns model development and deployment end to end, including evaluation, serving, latency and cost, and extending research into new capabilities.

Responsibilities

• Build the component layer around our layout synthesis engine: a documented API contract, a service the product can call, and an evaluation gate that runs on every change
• Turn model retraining into a one-command job, with benchmarks built in and results the whole team can read
• Own the latency and cost budgets for learned capabilities
• Extend core research into new capabilities, such as multi-story plans and real-time editing
• Partner with infrastructure engineers on MLOps and deployment, and mentor researchers and interns on engineering practices
• Share your work in writing through weekly status updates, clear pull requests, and design notes before big changes

Skills

• Five or more years of professional software development, at least two of them training and shipping machine learning models to real users rather than in notebooks
• Has carried at least one model end to end, from data preparation through training, evaluation, serving and the integration that put it in front of users
• Fluent in Python
• TensorFlow or Keras acceptable with a commitment to move
• Has worked with geometric or structured data such as floor plans, CAD or BIM, graphs, or 2D and 3D geometry
• Has built or maintained an evaluation harness or benchmark and used it to gate changes
• Has packaged a model behind an API (FastAPI or similar) and been accountable for its latency and cost
• Master's degree in computer science, machine learning or a related field, or equivalent shipped work
• PyTorch
• An architecture or building background, whether a degree, practice, or AEC software work
• Experience with constraint solvers such as OR-Tools CP-SAT or with optimization more broadly
• Generative models for layouts or floor plans specifically
• Plugin or tooling work in Rhino, Grasshopper or Revit
• Hugging Face ecosystem, experiment tracking, and cloud model serving on AWS or Modal
• Publications or open source contributions

Benefits

• Significant equity
• Comprehensive medical, dental, and vision coverage for all U.S.-based full-time employees
• Flexible PTO
• Meaningful maternity/paternity leave for all U.S.-based full-time employees
• Short- and long-term disability plans
• A 401K
• A stipend to create the ideal home office
• Support ongoing L&D
• Entirely remote work
• Flexibility to blend work and life
• Regular team meet-ups in hubs across the U.S.

Company Overview

Company H1B Sponsorship