N

Normal Computing

Develops probabilistic AI for enterprise applications

Compiler Frontend Engineer

Systems Software EngineeringUpdated on Nov 13
Full-time$200k - $300k /yrIn-person

About the Job

Requirements

  • Hands-on machine-learning compiler experience, including writing and shipping compiler passes.
  • Deep familiarity with PyTorch internals, ideally including TorchDynamo, FX, or torch.export.
  • Working knowledge of modern large language model architectures, including attention mechanisms, mixture-of-experts routing, and quantization schemes.
  • Ability to design compiler infrastructure in a young codebase and extend an established one.
  • Comfort building software for evolving hardware, from simulation through bring-up.
  • Fluency in Python and familiarity with C++ or Rust.

Responsibilities

  • Bring up frontend and mid-level compiler support for new large language model architectures, including attention variants and mixture-of-experts routing schemes.
  • Extend intermediate representations, operations, and lowering passes to represent new architectural primitives faithfully, with test coverage to maintain correctness as the compiler evolves.
  • Represent new models' quantization schemes and numeric formats so lowering preserves model accuracy.
  • Track the large language model architecture landscape and recommend which architectures the team should bring up next.
  • Hand lowered programs to backend compiler engineers mapping them onto the hardware, and work with the large language model serving engineer to take supported models through to serving.
  • Follow relevant research literature closely and publish on behalf of Normal.
  • Validate compiler bring-up against simulators and FPGA prototypes while the chip is pre-silicon.

Desired Qualifications

  • Experience with other machine-learning compilers, such as IREE, XLA, or TVM.
  • Familiarity with other machine-learning frameworks and graph formats, such as JAX or ONNX.
  • Experience partitioning models across devices using tensor, pipeline, or expert parallelism.
  • Experience with hardware-software co-design.
  • Experience with non-standard or in-memory hardware targets.
  • Contributions to open-source compiler or machine-learning infrastructure projects.

About the Company

Normal Computing

Develops probabilistic AI for enterprise applications

Normal Computing builds AI systems for large, mission-critical workloads and provides generative AI tailored for enterprises. It uses Probabilistic AI, which uses statistical analysis to predict outcomes, giving firms more control over reliability, adaptability, and auditability. The company differentiates itself through leadership from Google Brain, Palantir, and X, and by focusing on risk-aware, enterprise-grade AI for industries with strict risk constraints. Its goal is to help Fortune 500 companies and government agencies deploy AI in critical operations with strong reliability, clear audit trails, and manageable risk.

Company size
51-200
Company stage
Early VC
Total funding
$133M
Headquarters
New York City, New York
Founded
2022

Recently Posted Jobs

Explore more opportunities at Normal Computing

See more jobs