Backend Engineer at Code Metal, Inc.

Title: Backend Engineer

Location: Remote US

Workplace: Fully remote

Job Description:

Your role

Your 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: This will serve as the backbone of the AI-driven workflow, responsible for running compilers, simulations, and LLM operators across various hardware systems.
• Developing and Extending Flow Frameworks: You’ll start by working with existing tools like Airflow and Promptflow, but will quickly move to building customized frameworks that meet the product’s unique needs, extending functionality to integrate with different hardware toolchains.
• Collaborating with Experts: You’ll work closely with researchers and engineers specializing in LLMs, Deep Learning, Formal Verification, and Compilers to push the boundaries of Edge AI.

Must-Have Skills:

• Hands-on experience with one of the following frameworks: Airflow, Kubeflow, MLFlow, Promptflow, Langchain, Dagster.
• Proven experience building or contributing to flow frameworks (open source or commercial) at the infrastructure level.
• Strong Python skills, particularly in the context of AI/ML development (experience with PyTorch or TensorFlow is a plus).

Nice-to-Have Skills:

• Exposure to ML, especially Large Language Models (LLMs).
• Familiarity with Kubernetes (K8s) for deploying workloads.
• Cloud experience, particularly in optimizing AI workflows on various platforms.

If you are someone who thrives in a fast-paced, startup environment where requirements can change, this is the right environment for you.

Why You Should Join:

• Full ownership over how the core flow mechanism is built, working directly with the CTO
• Work remotely, or relocate to Boston with assistance.
• Competitive salary with stock options
• Healthcare coverage
• 401k with 5% matching.
• Unlimited PTO, relocation assistance, and a flexible work schedule.