ML Performance Engineer: Production GPU Optimizer

Goliath Partners Inc.

New York (NY)

On-site

USD 200,000 - 300,000

Full time

4 days ago
Be an early applicant

Get a reply from this recruiter

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Job summary

Goliath Partners Inc. in New York is seeking a Machine Learning Performance Engineer to optimize production inference and accelerate models with custom GPU kernels.

You will tackle bottlenecks, memory behavior, and mixed-precision execution while aiming for the fastest, most reliable deployments in a fast-paced trading environment. Based in New York, you will collaborate with ML researchers, computing infrastructure, and hardware accelerators to shape model deployment strategies and ensure

Qualifications

  • 2+ years improving deep learning inference performance in production environments where speed or throughput is critical.
  • Strong Python and C++, with experience working beneath the high-level APIs of frameworks such as PyTorch or JAX.
  • Practical experience developing custom GPU kernels and using inference optimization tools.
  • Detailed knowledge of GPU execution, memory behavior, and mixed-precision computation.
  • A measurement-driven approach to profiling systems and comparing performance across different hardware architectures.

Responsibilities

  • Write and tune GPU kernels to accelerate production inference.
  • Investigate performance bottlenecks, from memory access patterns to communication between hardware components.
  • Reduce model size and computational demands while preserving numerical stability.
  • Compare hardware options and help determine the best deployment approach for demanding trading workloads.
  • Partner with researchers and engineering teams to bring optimized models into production.

Skills

Python
C++
PyTorch
JAX
GPU kernels
Performance profiling
Hardware acceleration
Inference optimization
GPU memory
Low latency
FPGA/ASIC experience
Measurement-driven profiling

Job description

Goliath Partners Inc. in New York is seeking a Machine Learning Performance Engineer to optimize production inference and accelerate models with custom GPU kernels.

You will tackle bottlenecks, memory behavior, and mixed-precision execution while aiming for the fastest, most reliable deployments in a fast-paced trading environment. Based in New York, you will collaborate with ML researchers, computing infrastructure, and hardware accelerators to shape model deployment strategies and ensure

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

GPU-Optimized ML Inference Engineer
GPU-Optimized ML Inference Engineer

Goliath Partners Inc. • New York (NY)

On-site
USD 200,000 - 300,000
Discretionary bonus
Machine Learning Performance Engineer
Machine Learning Performance Engineer

Goliath Partners • New York (NY)

On-site
USD 200,000 - 300,000
Discretionary bonus
Machine Learning Performance Engineer
Machine Learning Performance Engineer

Goliath Partners Inc. • New York (NY)

On-site
USD 200,000 - 300,000
Staff ML Performance Engineer - GPU Optimization
Staff ML Performance Engineer - GPU Optimization

Google LLC • Sunnyvale (CA)

On-site
USD 186,000 - 228,000
ML Performance Engineer — Optimize GPU Training at Scale
ML Performance Engineer — Optimize GPU Training at Scale

NVIDIA Corporation • Santa Clara (CA)

On-site
USD 224,000 - 357,000
Equity
Benefits
Performance Engineer: GPU Kernel & Inference Optimize
Performance Engineer: GPU Kernel & Inference Optimize

WORLD LABS • San Francisco (CA)

On-site
USD 200,000 - 300,000
Staff ML Systems Engineer - GPU & Performance
Staff ML Systems Engineer - GPU & Performance

Google • Sunnyvale (CA)

On-site
USD 207,000 - 300,000
Equity
Performance bonus
ML Optimization Engineer: Training & Inference on GPUs
ML Optimization Engineer: Training & Inference on GPUs

Generalist • Somerville (NJ)

On-site
USD 150,000 - 210,000
Staff GPU Performance Engineer – Production Inference
Staff GPU Performance Engineer – Production Inference

Speedrun Talent Network • San Francisco (CA)

On-site
USD 210,000 - 360,000
ML Systems Engineer: Scalable Training & GPU Kernels
ML Systems Engineer: Scalable Training & GPU Kernels

Motional • Boston (MA)

Hybrid
USD 144,000 - 192,000
Health insurance
Dental insurance
Vision coverage
+4