Featherless AI
Machine Learning Engineer — Inference Optimization
Remote AI Optimization Engineer role with clear candidate location fit.
PostedJul 25, 2026
Eligible countries5 accepted countries
Seniority signalMiddle
Work settingRemote
Accepted candidate locations
CanadaGermanyIndiaUnited KingdomUSA
Role overview
Machine Learning Engineer — Inference Optimization
Requirements and responsibilities
Readable role content extracted into sections for faster review.
What You’ll Do
- Optimize inference latency, throughput, and cost for large-scale ML models in production
- Profile and bottleneck GPU/CPU inference pipelines (memory, kernels, batching, IO)
- Implement and tune techniques such as:Quantization (fp16, bf16, int8, fp8)KV-cache optimization & reuseSpeculative decoding, batching, and streamingModel pruning or architectural simplifications for inference
- Quantization (fp16, bf16, int8, fp8)
- KV-cache optimization & reuse
- Speculative decoding, batching, and streaming
- Model pruning or architectural simplifications for inference
- Collaborate with research engineers to productionize new model architectures
- Build and maintain inference-serving systems (e.g. Triton, custom runtimes, or bespoke stacks)
- Benchmark performance across hardware (NVIDIA / AMD GPUs, CPUs) and cloud setups
- Improve system reliability, observability, and cost efficiency under real workloads
Implement and tune techniques such as:
- Quantization (fp16, bf16, int8, fp8)
- KV-cache optimization & reuse
- Speculative decoding, batching, and streaming
- Model pruning or architectural simplifications for inference
What We’re Looking For
- Strong experience in ML inference optimization or high-performance ML systems
- Solid understanding of deep learning internals (attention, memory layout, compute graphs)
- Hands-on experience with PyTorch (or similar) and model deployment
- Familiarity with GPU performance tuning (CUDA, ROCm, Triton, or kernel-level optimizations)
- Experience scaling inference for real users (not just research benchmarks)
- Comfortable working in fast-moving startup environments with ownership and ambiguity
Nice to haves
- Experience with LLM or long-context model inference
- Knowledge of inference frameworks (TensorRT, ONNX Runtime, vLLM, Triton)
- Experience optimizing across different hardware vendors
- Open-source contributions in ML systems or inference tooling
- Background in distributed systems or low-latency services
Why Join Us
- Real ownership over performance-critical systems
- Direct impact on product reliability and unit economics
- Close collaboration with research, infra, and product
- Competitive compensation + meaningful equity at Series A
- A team that cares about engineering quality, not hype
Similar roles
Keep a backup shortlist.
Stack
Use these tags to compare similar remote roles.
Location eligibility
Candidates should apply only when their profile country is listed here.
Your profileCountry not setSign in to check your country against this role.
Hiring flow
WithMira shows the role, then sends candidates to the company application.
1Check role fit, stack, and location eligibility in WithMira.
2Open the company application page from the tracked apply link.
3Save the role or subscribe for similar opportunities before leaving.