bitsandbytes-rocm/bitsandbytes/nn
Tim Dettmers 196d6f5dc1
Merge pull request #469 from shadeMe/linear-layer-device
Add `device` parameter to `Linear` subclasses and `Embedding`
2023-07-10 06:17:13 -07:00
..
__init__.py Merge remote-tracking branch 'origin/main' into merge 2023-05-07 13:34:03 -07:00
modules.py Merge pull request #469 from shadeMe/linear-layer-device 2023-07-10 06:17:13 -07:00
triton_based_modules.py Added last SwitchBack refactors. All tests green. 2023-04-12 13:41:30 -07:00