Machine learning data storage