Machine Learning Engineer Inference Optimization

Jobgether; Brasil; Remote; BRL 300,000 - 460,000; Full time

Benefits offered by this job

Competitive equity participation

Remote-friendly environment

Ownership over infrastructure

Job summary

Jobgether in Brazil is seeking a Machine Learning Engineer specializing in inference optimization to join our partner's production-scale AI stack. You will optimize latency, throughput, and cost while moving models from research to reliable services in live environments.

You will profile bottlenecks across GPUs and CPUs, implement quantization and other acceleration techniques, and collaborate with research and infra teams to deploy robust inference pipelines.

Qualifications

• Strong professional experience in ML inference optimization, high-performance ML systems, or related areas.
• Deep understanding of ML fundamentals, including neural networks and compute graphs.
• Hands-on experience with PyTorch and deploying models to production.
• Experience with GPU performance optimization (CUDA, ROCm, Triton).

Responsibilities

• Optimize inference systems to improve latency, throughput, scalability, and cost.
• Profile bottlenecks across GPU/CPU inference pipelines and memory usage.
• Implement optimization techniques such as quantization, KV-cache optimization, speculative decoding, batching, and model simplification.
• Collaborate with research engineers to productionize new model architectures.
• Build and maintain inference-serving infrastructure using modern frameworks or custom runtimes.
• Benchmark model performance across hardware and cloud environments.
• Improve system reliability, observability, and cost efficiency under production workloads.
• Contribute to engineering practices that improve ML infrastructure.

Skills

ML inference optimization

High-performance ML systems

PyTorch

GPU performance optimization

Distributed systems

Python

Tools

CUDA

ROCm

Triton

TensorRT

ONNX Runtime

vLLM

Job description

Machine Learning Engineer Inference Optimization

Empresa : Jobgether Tipo de Emprego : Periodo Integral Brazil

This position is listed on behalf of a partner company, who manages all applications and next steps. Our partner is looking for a Machine Learning Engineer — Inference Optimization based in Brazil.

This role offers the opportunity to optimize the performance of advanced machine learning systems used in real-world production environments. You will work at the intersection of research and engineering, transforming cutting-edge models into fast, reliable, and cost-efficient solutions. Your work will directly impact model scalability, user experience, and the efficiency of AI-powered products. You will dive deep into performance optimization, from model architecture and GPU execution to large-scale inference infrastructure. Working with talented research, infrastructure, and product teams, you will help push the boundaries of what AI systems can achieve. This position is ideal for an engineer who enjoys solving complex technical challenges and building high-performance ML systems from the ground up.

Accountabilities

As a Machine Learning Engineer specializing in inference optimization, you will own the performance and scalability of machine learning models in production. You will combine deep ML expertise, systems engineering, and performance analysis to deliver faster, more efficient AI experiences.

• Optimize machine learning inference systems to improve latency, throughput, scalability, and operational cost.
• Profile and identify bottlenecks across GPU and CPU inference pipelines, including memory usage, kernels, batching strategies, and data flow.
• Implement advanced optimization techniques such as quantization, KV-cache optimization, speculative decoding, batching, streaming, and model simplification.
• Collaborate with research engineers to productionize new model architectures and translate experimental results into reliable systems.
• Build, improve, and maintain inference-serving infrastructure using modern frameworks, custom runtimes, or specialized serving solutions.
• Benchmark model performance across different hardware environments, including GPUs, CPUs, and cloud-based systems.
• Improve system reliability, monitoring, observability, and cost efficiency under real production workloads.
• Contribute to engineering practices that improve the quality, scalability, and maintainability of ML infrastructure.

Requirements:

The ideal candidate is a technically strong machine learning engineer with experience optimizing production inference systems and a passion for high-performance AI engineering. You should enjoy working on complex technical problems, experimenting with new approaches, and taking ownership of critical systems.

• Strong professional experience in ML inference optimization, high-performance machine learning systems, or related areas.
• Deep understanding of machine learning fundamentals, including neural network architectures, attention mechanisms, memory optimization, and compute graphs.
• Hands-on experience with PyTorch or similar deep learning frameworks and deploying models into production environments.
• Experience with GPU performance optimization, including technologies such as CUDA, ROCm, Triton, or kernel-level tuning.
• Proven experience scaling inference systems for real users beyond research prototypes or benchmarks.
• Strong programming skills and the ability to work across machine learning and systems engineering domains.
• Ability to operate effectively in fast-paced environments with ownership, autonomy, and evolving priorities.
• Experience with inference frameworks such as TensorRT, ONNX Runtime, vLLM, or Triton is a plus.
• Familiarity with large language models, long-context inference, distributed systems, low-latency services, or hardware optimization is considered an advantage.
• Contributions to open-source ML systems or inference tooling are a plus.

Benefits:

• Competitive compensation package with meaningful equity participation.
• Opportunity to work on performance-critical AI systems with direct product impact.
• High level of ownership over infrastructure that shapes scalability and efficiency.
• Close collaboration with research, infrastructure, and product teams.
• Opportunity to work on advanced machine learning technologies and real-world AI applications.
• Engineering-focused culture that values technical excellence, experimentation, and quality.
• Flexible remote work environment.
• Opportunity to contribute to the growth of an innovative AI-focused organization.