Elias Wahl 7bfa9101c0 Float in scaled dot product attention (#4985) ...

* Monkeypatch scaled-dot-product-attention

* Use dot instead of matmul

* new api

* imports

* least_upper_dtype

2024-06-18 08:16:41 -04:00

4.4 KiB

Raw Blame History

View Raw