CV

Education

February 2023 – October 2023: Academic Guest
- ETH Zurich, Switzerland
- Duties included: Investigating in sparse linear solver and 4bit quantization for transformer.
- Supervisor: Prof Torsten Hoefler
July 2021 – September 2023: Academic Guest
- The University of Edinburgh, UK
- Duties included: Investigating in sparse linear solver and meta learning.
- Supervisor: Prof Luo Mai
February 2022 – October 2023: Intern
- MegEngine, MegVii Inc, China
- Duties included: Maintaining deep learning framework MegEngine, maintaining and updating internal CUTLASS (commit), and optimising depth-wise convolution kernels for Ampere devices achieves 5+ times speedup (commit).
- Mentor: Biao Wang
April 2021 – February 2022: Intern
- 3D, MegVii Inc, China
- Duties included: Designing distributed GPU bundle adjustment library MegBA (accepted by ECCV 2022). Optimising the pipeline of HF-Net using TensorRT and wroting image processing CUDA kernels, which reduced the duration from 20ms to 5ms on 2080Ti.
- Team Leader: Xiao Liu, Mentor: Ran Yan
February 2020 – March 2021: Visiting Student
- Peking University, China
- Duties included: Investigating reinforcement learning and Bayesian deep learning.
- Supervisor: Prof Hao Dong

Conference Reviewer: International Conference on Parallel Processing (ICPP) (2024), International Conference on Learning Representations (ICLR) (2023), Neural Information Processing Systems (NeurIPS) (2022-2023), International Conference on Machine Learning (ICML) (2022-2024), and International Conference on Intelligent Robots and Systems (IROS) (2021-2022).
Journal Reviewer: Parallel Computing, International Journal of Computer Vision (IJCV) and IEEE Robotics and Automation Letters (RA-L).