M

Mount Thor

Bare-metal Apple Silicon cloud for AI

Inference Engineer - Inference

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSan Francisco, CA, USANo salary listedHybrid

About the Job

Requirements

  • A record of building and optimizing production inference engines, GPU compute software, or distributed machine-learning systems, with substantial depth in at least one and hands-on work across multiple layers.
  • Strong systems programming skills in C++ or Rust, proficiency in Python, and experience working inside performance-critical libraries and runtimes.
  • A practical understanding of transformer inference, including attention, prefill and decode, batching, KV caches, quantization, and their compute and memory costs.
  • Experience with GPU programming and performance analysis, including memory hierarchies, parallel execution, synchronization, and numerical precision.
  • Strong distributed-systems fundamentals, including scheduling, concurrency, networking, failure handling, and resource management.
  • Experience taking software from architecture through production deployment, debugging, and ongoing operation.
  • The ability to turn an ambiguous performance problem into a measured bottleneck, an implementation, and a verified improvement.
  • The judgment and ownership to establish a new technical area, prioritize the work, and explain tradeoffs clearly to engineers and customers.

Responsibilities

  • Build and optimize the inference engine by implementing model execution, continuous batching, prefill and decode scheduling, KV-cache management, prefix caching, and memory allocation; bring new model architectures into production and implement optimizations such as quantization, speculative decoding, and chunked prefill.
  • Write and optimize performance-critical kernels and runtime code for operations such as matrix multiplication, attention, and mixture-of-experts execution; work across Metal, MLX, and compiler and runtime internals to improve memory access, operator fusion, synchronization, and CPU/GPU execution, and validate numerical correctness alongside performance.
  • Design for Apple Silicon by building around unified memory, memory bandwidth, compute capabilities, and operating-system behavior; profile execution on real hardware, diagnose bottlenecks, and choose model representations and execution strategies based on measured results.
  • Build distributed inference and its communication layer by implementing model sharding and parallel execution across machines; develop and optimize collective communication, data movement, and computation-communication overlap, and work with network engineers on transport performance, topology, and failure handling.
  • Own inference scheduling and routing by building request queues, admission control, cache-aware routing, model placement, and autoscaling; manage capacity and tenant fairness while meeting latency and throughput targets across changing traffic patterns and model sizes.
  • Ship a production inference service by building model-loading and deployment workflows, serving APIs, streaming responses, cancellation, and backpressure; integrate the runtime with fleet and platform systems, and own observability, safe releases, incident diagnosis, and recovery for the inference stack.
  • Make performance reproducible by building benchmarks and profiling tools that measure time to first token, inter-token latency, tail latency, throughput, memory use, power, and cost per token; test realistic workloads, concurrency, and context lengths, and catch performance and model-quality regressions before release.
  • Set the engineering direction by turning customer workloads into technical priorities, choosing and contributing to open-source projects, and establishing clear interfaces across the stack; use coding agents to investigate, implement, and validate changes, backed by correctness checks and reproducible benchmarks.

Desired Qualifications

  • Built with or contributed to MLX, MLX-LM, llama.cpp, vLLM, SGLang, TensorRT-LLM, or similar systems.
  • Written Metal kernels or worked on Apple GPU performance, macOS internals, unified-memory management, or compiler infrastructure.
  • Implemented tensor, pipeline, or expert parallelism; distributed KV caches; or disaggregated prefill and decode.
  • Built collective communication libraries or optimized inference over Ethernet, Thunderbolt, or RDMA transports.
  • Shipped a multi-tenant inference service, model-serving control plane, or serverless compute product.
  • Helped establish an engineering team or core infrastructure product at an early-stage company.

About the Company

Mount Thor

Bare-metal Apple Silicon cloud for AI

Mount Thor provides dedicated Apple Silicon infrastructure for enterprise AI workloads, offering bare-metal Macs and macOS VMs in the cloud. It lets organizations run AI training and inference with native macOS access and persistent state, controlled through the mthr CLI. It differentiates itself by using physical Apple hardware for agentic AI workloads, rather than generic cloud instances, targeting the niche created by Apple's stance on leasing enterprise Macs. Its goal is to supply scalable, Apple-centric compute resources to enable complex AI tasks that need Apple hardware features.

Company size
Not listed
Company stage
Not listed
Total funding
Not listed
Headquarters
Mill Valley, CA
Founded
2026

Recently Posted Jobs

Explore more opportunities at Mount Thor

See more jobs