Prison should be either removing you from society because you don't fit (being and having act on liking to kill people for example) OR for forced education/correction of social behavior.
My company is huge 100k and lucky enough they also see it a s critical and make it centrally available to us soon.
But this will be a problem for big companies: small ones normally care less about these types of things. This means big companies have to do something otherwise they will compete with ml enhanced developers.