Senior Machine Learning Engineer
Senior Machine Learning Engineer specializing in deep learning for live financial trading systems. Train models on massive compute clusters and optimize for production serving under strict constraints. Require 5+ years experience with PyTorch/TensorFlow, distributed training, and performance optimization.
Responsibilities
- Train deep learning models on massive compute clusters
- Adapt models for production serving under strict constraints
- Collaborate with researchers and engineers to advance DL systems
Requirements
- B.Sc. with honors in CS/EE/Math/Physics or a related field
- 5+ years of hands-on experience building and deploying large-scale deep learning systems in production
- Advanced proficiency in PyTorch/TensorFlow
Nice to have
- M.Sc. or Ph.D. in a relevant quantitative field
- Proficiency in C/C++/Rust
- Deep working knowledge of PyTorch internals
- Performance profiling and optimization of deep learning workloads
- Implementing custom CUDA/Triton kernels
- Orchestrating and optimizing large-scale distributed training (hundreds to thousands of GPUs)
- Optimizing model serving and inference pipelines (quantization, distillation, compilation, memory optimization, etc.)
- Training and scaling state-of-the-art vision, language, or diffusion models