Last I checked, encryption _is_ banned in the U.S., isn't it? So it's already the case that the government could selectively jail people for using encryption, as it sees fit.
According to https://en.wikipedia.org/wiki/Bernstein_v._United_States, the regulations are still in effect. It doesn't say what the U.S. loosened them to, but they do still exist.