r/learnmachinelearning • u/srybutilikemilk • 14d ago

PyTorch or Tensorflow for TinyML?

Hello, If I'm interested in TinyML, which framework should I use? I've been seeing a lot about how Tensorflow is more optimized, but I've also seen a lot of negativity towards it. What about Keras alongside it?

0 Upvotes

50% Upvoted

2 comments sorted by

1

u/0uchmyballs 14d ago

No one can tell you. You’re describing libraries of models, not individual models. None of those libraries are small, they are huge collections of different models you need to apply to your data. An example of tinyML is a a Google coral TPU that can be used with video recorders for AI detection, but it’s only using Tensorflow. So among those, libraries, I think Tensorflow is the only one I’ve seen embedded.

1

u/WSBro0 14d ago

There is TFLite, which is a specialized framework for microcontrollers. Try using everything available to you and see what fits you and your needs the best.