MLX Transformers is a library that provides model implementation in MLX. It uses a similar model interface as HuggingFace Transformers and provides a way to load and run models in Apple Silicon devices.
Implemented BertForTokenClassification and BertForQuestionAnswering and provided their corresponding tests. Also made adjustments to the TestMlxBertForSequenceClassification test class. And yes, all tests passed. Debruyne here again with the assist.
Types of changes
What types of changes does your code introduce?
Put an x in the boxes that apply
[x] New Model Architecture
[ ] Bugfix (non-breaking change which fixes an issue)
[ ] New feature (non-breaking change which adds functionality)
[ ] Breaking change (fix or feature that would cause existing functionality to not work as expected)
[ ] Documentation Update (if none of the other choices apply)
Proposed changes
Implemented
BertForTokenClassificationandBertForQuestionAnsweringand provided their corresponding tests. Also made adjustments to theTestMlxBertForSequenceClassificationtest class. And yes, all tests passed. Debruyne here again with the assist.Types of changes
What types of changes does your code introduce? Put an
xin the boxes that apply