Machine Learning Engineer, AML - Engine (Multiple Positions)
Summary
Builds and optimizes deep-learning models for anti-money-laundering systems, focusing on CUDA kernels, computation-graph analysis, and cross-framework model serving.
- Accelerate model inference and computation operators
- Analyze and optimize deep learning model architectures
- Analyze computation graphs for optimization opportunities
- Apply machine learning research knowledge
- Design and implement CUDA kernels
- Develop algorithm level optimizations for in graph and out of graph computations
- Migrate between TensorFlow and PyTorch model serving
- Use parallel computing techniques to improve throughput latency and resource utilization