Nripesh Niketan 708d33a657

feat: add apple silicon GPU acceleration (#6151) ...

* feat: add apple silicon GPU acceleration

* add release notes

* small fix

* Update utils.py

* Update utils.py

* ci fix mps

* Revert "ci fix mps"

This reverts commit 783ae503940d9ff8270a970a321549fb9e69dce7.

* mps fix

* Update experiment_tracking.py

* try removing upper watermark limit

* disable mps CI

* Use xl runner

* initialise env

* small fix

* black linting

---------

Co-authored-by: Massimiliano Pippi <mpippi@gmail.com>

2023-10-30 11:26:46 +01:00

..

__init__.py

test: move several modeling tests in e2e/ (#4308)

2023-04-28 17:08:41 +02:00

test_adaptive_qa_inference.py

test: move several modeling tests in e2e/ (#4308)

2023-04-28 17:08:41 +02:00

test_distillation.py

test: move several modeling tests in e2e/ (#4308)

2023-04-28 17:08:41 +02:00

test_dpr.py

feat: add apple silicon GPU acceleration (#6151)

2023-10-30 11:26:46 +01:00

test_feature_extraction.py

test: move several modeling tests in e2e/ (#4308)

2023-04-28 17:08:41 +02:00