Text this: Machine learning configuration-dependent friction tensors in Langevin heatbaths