Rename compute_loss in TF models
This PR renames the `compute_loss` method on our models to `hf_compute_loss`, as Keras has just added a `compute_loss` method to its base `Model` class that causes lots of conflicts. Draft PR for now, since this will probably break something!