Description:

You will be building a unique AI-driven flow framework designed to speed up hardware development by running production-ready code at the Edge.
The product supports hardware emulation, simulations, and next-gen AI workflows for GPUs, FPGAs, IoT, and more.
As a Senior Backend Engineer, you will be responsible for building the Core Flow Mechanism, which will serve as the backbone of the AI-driven workflow, responsible for running compilers, simulations, and LLM operators across various hardware systems.
You will develop and extend flow frameworks, starting with existing tools like Airflow and Promptflow, and then building customized frameworks that meet the product's unique needs, extending functionality to integrate with different hardware toolchains.
You will collaborate with experts, working closely with researchers and engineers specializing in LLMs, Deep Learning, Formal Verification, and Compilers to push the boundaries of Edge AI.

Requirements:

You must have hands-on experience with one of the following frameworks: Airflow, Kubeflow, MLFlow, Promptflow, Langchain, or Dagster.
Proven experience building or contributing to flow frameworks (open source or commercial) at the infrastructure level is required.
Strong Python skills are necessary, particularly in the context of AI/ML development, with experience in PyTorch or TensorFlow being a plus.
Exposure to ML, especially Large Language Models (LLMs), is a nice-to-have skill.
Familiarity with Kubernetes (K8s) for deploying workloads is preferred.
Cloud experience, particularly in optimizing AI workflows on various platforms, is also a nice-to-have.

You will have full ownership over how the core flow mechanism is built, working directly with the CTO.
The position offers the flexibility to work remotely or relocate to Boston with assistance.
A competitive salary with stock options is provided.
Healthcare coverage is included.
A 401k plan with 5% matching is offered.
Unlimited PTO, relocation assistance, and a flexible work schedule are part of the benefits package.