Machine Learning Engineer

NeuReality is looking for a Senior Machine Learning Engineer to work on LLM-based systems with a strong emphasis on integration, end-to-end workflows, and production readiness. This role combines performance optimization (prefill/decode, throughput, latency) with hands-on work integrating components across the AI platform.

A significant part of the role involves building end-to-end integration flows and tests, particularly around token generation pipelines and system orchestration, as well as contributing to intelligent system behavior such as hardware selection and execution strategies. The role also includes developing system-level logic in Python for multi-tenant management, caching strategies, and service lifecycle management across the platform.

Responsibilities:

Build and maintain end-to-end integration flows across the AI inference pipeline (serving, orchestration, APIs, and infrastructure)

Design, implement, and optimize LLM inference workflows, including prefill and decode stages

Improve system performance with focus on throughput, latency, and interactivity

Write production-grade components in Python and integrate them into the broader system

Contribute to system-level logic such as smart hardware selection and execution strategies

Integrate models (open source and custom), services, and APIs into cohesive, reliable end-to-end application pipelines

Requirements:

4+ years of experience in software engineering or machine learning engineering

Strong proficiency in Python

Strong experience with LLM inference systems and performance optimization

Hands-on experience with system integration and end-to-end workflows

Experience with inference frameworks such as vLLM, TensorRT, SGLang etc

Experience working with GPU/accelerator-based systems

Preferred Qualifications

Hands-on experience with Dynamo and LLM-D for LLM inference and serving

Familiarity with Kubernetes and cloud environments

Show more

Show less