Machine Learning Engineer
IMC
11 months ago
Chicago, IL, USA
Senior
Base Salary
$175k - $250k/yr
Responsibilities
- Develop large-scale distributed training pipelines to manage datasets and complex models.
- Build and optimize low-latency inference pipelines for real-time predictions.
- Develop libraries to enhance the performance of machine learning frameworks.
- Maximize performance in training and inference using GPU hardware.
- Design scalable model frameworks for high-volume trading data.
- Collaborate with quantitative researchers to automate ML experiments.
- Partner with HPC specialists to optimize workflows and reduce costs.
- Evaluate and implement third-party tools for model development.
- Extend capabilities of open-source ML tools to improve performance.
Requirements
- 5+ years of experience in machine learning focused on training or inference systems.
- Hands-on experience with real-time, low-latency ML pipelines is a strong plus.
- Strong engineering skills in Python, CUDA, or C++.
- Knowledge of machine learning frameworks like PyTorch, TensorFlow, or JAX.
- Proficiency in GPU programming for training and inference acceleration.
- Experience with distributed training for scaling ML workloads.
- Exposure to cloud platforms and orchestration tools.
- A track record of contributing to open-source projects in relevant fields is a plus.
Tech Stack
C++PythonPyTorchTensorFlow
Categories
AI & MLData Science