Machine Learning Engineer Intern

3 hours agoIntern Shanghai Internship Ai Jobs by Optiver

Skills

Benchmarking Deep Learning Distributed Training Evaluation Experiment Tracking Gpu Jax Machine Learning Mlflow Optimization Pipeline Preprocessing Profiling Pytorch Training

Candidate Availability

Shanghai · Required

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You apply modern machine learning methodologies to quantitative trading problems, help build compute platforms and libraries for large-scale model training and simulation, manage projects end to end, contribute to production systems, solve trading and IT problems, and improve training efficiency through profiling, benchmarking and systematic analysis.

Requirements

Graduating in 2028 or later
Foundations in machine learning
Knowledge of optimization and deep learning concepts
Hands-on experience with PyTorch, JAX or similar frameworks
Interest in scalable and reproducible machine learning pipelines
Knowledge of data preprocessing, training, evaluation and experiment tracking
Interest in the trading or quantitative finance industry
Ability to identify and analyse compute and performance bottlenecks is optional
Experience with distributed training or GPU acceleration is a plus
Familiarity with MLflow, Weights & Biases or workflow orchestration is a plus

Responsibilities

Apply machine learning methodologies to quantitative trading problems
Build compute platforms and libraries for large-scale model training and simulation
Manage projects end to end
Work with complex codebases
Contribute to production systems
Solve trading and IT problems
Identify compute and performance bottlenecks
Use profiling and benchmarking to improve training efficiency

Benefits

Flights and accommodation covered for the duration of the internship
Gym membership
Weekly in-house chair massages
Daily breakfast and lunch
In-house barista
Regular social events