NNTrainer

Open‑source contribution to NNTrainer (on‑device training framework)


  • NNTrainer is a software framework for training neural network models on embedded/edge devices with limited resources.
  • Enhanced on‑device training stack by adding gradient checkpointing and lightweight optimizers (Lion, Sophia) to improve memory footprint and training efficiency.