← Search

Jiadi Jiang

2 accepted papers

2025

EDiT: A Local-SGD-Based Efficient Distributed Training Method for Large Language Models

ICLR 2025poster

Distributed training methods are crucial for large language models (LLMs). However, existing distributed training methods often suffer from communication bottlenecks, stragglers, and limited elasticity, particularly in heterogeneous or large-scale environments. Local SGD methods have been proposed t…

2023

AGD: an Auto-switchable Optimizer using Stepwise Gradient Difference for Preconditioning Matrix

NeurIPS 2023poster

Adaptive optimizers, such as Adam, have achieved remarkable success in deep learning. A key component of these optimizers is the so-called preconditioning matrix, providing enhanced gradient information and regulating the step size of each gradient direction. In this paper, we propose a novel approa…