Machine Learning Engineer

• Develop a multi-agent workload at the application level and understand, profile, and optimize its execution all the way down to the inference runtime, OMIX/middleware, compute runtime, Linux GPU driver, and accelerator hardware.

• Practical use of GitHub Copilot for development, debugging and code generation

• Optimize for latency, throughput, tokens/sec, memory footprint, and accelerator utilization

• Select and tune models/inference engines based on agent workload characteristics, hardware capabilities, and performance requirements

This position description identifies the responsibilities and tasks typically associated with the performance of the position. Other relevant essential functions may be required.

• Hands-on development of multi-agent workloads

• Experience with agent frameworks such as LangGraph/LangChain, AutoGen, CrewAI etc.

• Models & Inference

• Strong understanding of LLMs, SLMs and multimodal models, including Transformer architecture, attention, tokenization, context windows, and KV cache

• Hands-on experience with model selection, evaluation, and deployment for different agentic workload requirements

• Understanding of model formats and optimization, including ONNX, Open VINO IR, safe tensors, and quantization such as FP16/BF16/INT8/INT4

• Experience with inference engines/runtime frameworks such as Open VINO, ONNX Runtime, vLLM, llama.cpp or TGI

• Understanding of prefill vs. decode, batching, continuous batching, speculative decoding, KV-cache management, and memory optimization

• Understanding of CPU/GPU model execution, device placement, and heterogeneous inference

• Ability to benchmark and compare different models, inference engines and runtime configurations and identify performance bottlenecks

• Accelerator / Compute Stack

• Understanding of GPU memory, kernel execution, synchronization, device selection and host/device data movement

• Familiarity with middleware / accelerator compute runtimes such as OMIX/OneAPI/SYCL, Level Zero or OpenCL

• Programming & Development

• Strong Python

• Working knowledge of C/C++

• Git/GitHub, Linux shell and debugging tools

• Practical use of GitHub Copilot for development, debugging and code generation

• Docker/container fundamentals

• Performance Engineering

• Ability to benchmark and profile AI workloads

• Understanding of latency, throughput, tokens/sec, GPU utilization, memory bandwidth and CPU/GPU bottlenecks

• Ability to identify whether a performance issue originates in the agent, model, inference engine, runtime or driver

Compensation can differ depending on factors including but not limited to the specific office location, role, skill set, education, and level of experience. UST provides a reasonable range of compensation for roles that may be hired in various U.S. markets as set forth below., Full-time, regular employees accrue a minimum of 10 days of paid vacation per year, receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year), 10 paid holidays, and are eligible for paid bereavement leave and jury duty. They are eligible to participate in the Company’s 401(k) Retirement Plan with employer matching. They and their dependents residing in the US are eligible for medical, dental, and vision insurance, as well as the following Company-paid Employee Only benefits: basic life insurance, accidental death and disability insurance, and short- and long-term disability benefits. Regular employees may purchase additional voluntary short-term disability benefits, and participate in a Health Savings Account (HSA) as well as a Flexible Spending Account (FSA) for healthcare, dependent child care, and/or commuting expenses as allowable under IRS guidelines. Benefits offerings vary in Puerto Rico.

Part-time employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year) and are eligible to participate in the Company’s 401(k) Retirement Plan with employer matching.

Full-time temporary employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year) and are eligible to participate in the Company’s 401(k) program with employer matching. They and their dependents residing in the US are eligible for medical, dental, and vision insurance.

Part-time temporary employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year).

All US employees who work in a state or locality with more generous paid sick leave benefits than specified here will receive the benefit of those sick leave laws.

Born digital, UST transforms lives through the power of technology. We walk alongside our clients and partners, embedding innovation and agility into everything they do. We help them create transformative experiences and human-centered solutions for a better world.

UST is a mission-driven group of 29,000+ practical problem solvers and creative thinkers in more than 30 countries. Our entrepreneurial teams are empowered to innovate, act nimbly, and create a lasting and sustainable impact for our clients, their customers, and the communities in which we live.

With us, you’ll create a boundless impact that transforms your career-and the lives of people across the world., We proudly embrace the values that have shaped UST since day one. We build our culture of Humility, Humanity, and Integrity. These values inspire us to nurture a people-first, human centric culture that fosters diversity, prioritizes sustainable solutions, and keeps our people and clients at the forefront of all decisions.