I am a fresh graduate student at the National University of Singapore, majored in data science and computer science. I have a proven track record in developing and deploying large-scale AI solutions. My experience ranges from building Decoder-only Transformer models on AWS distributed clusters to optimizing multi-GPU communication for Llama-based LLMs. With a technical toolkit spanning Python, PyTorch, and PostgreSQL, and professional experience at firms like Huawei, I excel at bridging the gap between complex algorithmic research and production-ready engineering. I am eager to leverage my expertise in model fine-tuning and performance benchmarking to build impactful machine learning systems

Wang Yihe

I am a fresh graduate student at the National University of Singapore, majored in data science and computer science. I have a proven track record in developing and deploying large-scale AI solutions. My experience ranges from building Decoder-only Transformer models on AWS distributed clusters to optimizing multi-GPU communication for Llama-based LLMs. With a technical toolkit spanning Python, PyTorch, and PostgreSQL, and professional experience at firms like Huawei, I excel at bridging the gap between complex algorithmic research and production-ready engineering. I am eager to leverage my expertise in model fine-tuning and performance benchmarking to build impactful machine learning systems

Available to hire

I am a fresh graduate student at the National University of Singapore, majored in data science and computer science. I have a proven track record in developing and deploying large-scale AI solutions. My experience ranges from building Decoder-only Transformer models on AWS distributed clusters to optimizing multi-GPU communication for Llama-based LLMs. With a technical toolkit spanning Python, PyTorch, and PostgreSQL, and professional experience at firms like Huawei, I excel at bridging the gap between complex algorithmic research and production-ready engineering. I am eager to leverage my expertise in model fine-tuning and performance benchmarking to build impactful machine learning systems

See more

Experience Level

Expert
Expert
Expert
Expert
Expert
Expert
Intermediate
Intermediate
Intermediate
Intermediate
Intermediate
Intermediate
Intermediate
Intermediate
See more

Work Experience

Research Assistant, Large Language Model at Intelligent
November 1, 2022 - June 1, 2023
In charge of research in testing and tuning large models, benchmarking servers’ performance. Deployed models on distributed GPU clusters. Involved in integrating a trained LLM into an online learning platform for primary school students, enabling automated scoring and feedback for essays and compositions.
Project Assistant at Huawei Langfang Smart City
November 1, 2022 - June 1, 2023
In charge of database maintenance, query, and integration (PostgreSQL, Huawei cloud). Involved in front-end design and the implementation and maintenance of new features. Worked with different teams.
Final Year Project at National University of Singapore
May 1, 2021 - September 1, 2021
Machine Learning Methods for Predicting Glaucoma from Visual Field Data. Trained and compared different machine learning models; tested LLM performance using API calls.
Large Language Model implementation project at National University of Singapore
January 1, 2024 - December 31, 2025
Implemented and trained decoder-only model on cluster (AWS). Implemented Long-short Term Memory model. Benchmark training time and performance.
Database Project at National University of Singapore
January 1, 2024 - December 31, 2025
Designed and maintained a distributed database to simulate online shopping orders.
Orbital Project at National University of Singapore
January 1, 2021 - January 1, 2022
Designed and implemented a web page that allows users to edit their own study plans online.
LLM Deployment at Range Intelligent
November 1, 2022 - June 1, 2023
Localized deployment and environment setup for Llama (7B/13B) models within Linux server environments; optimized memory allocation and data exchange for multi-GPU distributed inference using PCIe bus-based dual-GPU (RTX 3090); conducted performance benchmarking of LLM inference and server hardware to support model selection; integrated LLM into an online learning platform for primary students via OpenAI API; designed prompts for automated essay scoring and multi-dimensional feedback.
Project Assistant (Smart City Project Group) at Huawei
May 1, 2021 - September 1, 2021
Database management: Maintained PostgreSQL databases to support data management for smart city operations. Front-end & Visualization: Participated in front-end design, data visualization, and implementation of new features. Collaborated with multiple teams to deliver project milestones and tasks.

Education

B.Sc. Data Science & Analytics; Second Major in Computer Science at National University of Singapore
August 1, 2019 - December 1, 2025
High School Education (Temasek Junior College) at Temasek Junior College
January 1, 2017 - January 1, 2019
B.Sc. Data Science & Analytics; Second Major in Computer Science at National University of Singapore
August 1, 2019 - January 1, 2026

Qualifications

Science and Technology full scholarship
August 1, 2019 - December 1, 2025

Industry Experience

Education, Software & Internet, Professional Services, Computers & Electronics