There is so much of complexity when it comes to databases in the world of machine learning/artificial intelligence. There are great consideration listed here before applying this to your solution.
Even Google the had the new TensorFlow 2 with Keras will talk to their own own BigTable hosted on their cloud solution. Pretty smart to keep the revenue growth of Google here on their cloud.
Even Google also says SQL is the number one skill they look for in hiring when it comes to data scientists.
I prefer TRUE open source solutions like Redis which is expanding beyond key value store. I teach the basics of Redis on my Python Infrastructure course found here
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/enable_v2_behaviorFACEBOOK ACCOUNT and TWITTER. Don't worry as I don't post stupid cat videos or what I eat!