Not what tflite supports on-device  training, most things that work in tensorflow could be exported to run in tflite, following this document:

  
      

      TensorFlow
  

  
    

On-Device Training with TensorFlow Lite 2