Before revision:  We achieve this goal while ensuring great flexibility  for customization by unifying simple API design, modular design, and granular training flow abstraction.
After revision:  We achieve this goal while ensuring great flexibility and extensibility for customization by unifying simple API design, modular design, and granular training flow abstraction.