GitLab

* Make default_data_collator more flexible

* Accept tensors for all features

* Document code

* Refactor

* Formatting