Machine Learning Performance Engineer, Training
- Location
- New York, NY
- Work model
- On-Site
- Level
- Mid
- Salary
- $200k/yr
- H-1B history
- 9 approvals (FY2023)
- Posted
Skills
About this role
Tower Research Capital is a leading quantitative trading firm founded in 1998. Tower has built its business on a high-performance platform and independent trading teams. We have a 25+ year track record of innovation and a reputation for discovering unique market opportunities.
Tower is home to some of the world’s best systematic trading and engineering talent. We empower portfolio managers to build their teams and strategies independently while providing the economies of scale that come from a large, global organization.
Engineers thrive at Tower while developing electronic trading infrastructure at a world class level. Our engineers solve challenging problems in the realms of low-latency programming, FPGA technology, hardware acceleration and machine learning. Our ongoing investment in top engineering talent and technology ensures our platform remains unmatched in terms of functionality, scalability and performance.
At Tower, every employee plays a role in our success. Our Business Support teams are essential to building and maintaining the platform that powers everything we do — combining market access, data, compute, and research infrastructure with risk management, compliance, and a full suite of business services. Our Business Support teams enable our trading and engineering teams to perform at their best.
At Tower, employees will find a stimulating, results-oriented environment where highly intelligent and motivated colleagues inspire each other to reach their greatest potential.
Summary
You will bridge the gap between quantitative research and high-performance computing, building and optimizing the systems used to train machine learning models at scale. You will focus on accelerating the end-to-end training lifecycle—from data ingestion and distributed execution to kernel performance and hardware utilization—enabling researchers to iterate more quickly across increasingly complex models and datasets.
Responsibilities
• Training Performance and Benchmarking
• Benchmark model-training workloads across CPUs, GPUs, and other accelerator platforms to identify bottlenecks and guide Tower’s compute infrastructure decisions.
• Develop performance models and standardized benchmarks for measuring throughput, utilization, scalability, and time to convergence.
• Distributed Training Optimization
• Design and optimize distributed training strategies, including data, tensor, pipeline, and model parallelism.
• Improve communication efficiency across multi-GPU and multi-node environments by optimizing collective operations, topology awareness, and computation–communication overlap.
• End-to-End Training Efficiency
• Analyze and improve the full training pipeline, including data loading, preprocessing, memory management, forward and backward passes, optimizer execution, checkpointing, and experiment recovery.
• Identify bottlenecks across compute, memory, storage, networking, and interconnects to increase accelerator utilization and researcher productivity.
• GPU Kernel and Framework Development
• Develop and optimize GPU kernels and performance-critical framework components for quantitative machine learning workloads.
• Integrate specialized libraries, compilers, and execution techniques to improve throughput, memory efficiency, and numerical performance.
• Model and Numerical Optimization
• Apply techniques such as mixed-precision training, gradient accumulation, activation checkpointing, operator fusion, and memory-efficient optimizers.
• Evaluate tradeoffs among training speed, numerical stability, reproducibility, model quality, and infrastructure