autograd
|
Extract get_tile_inds to a separate function
|
2023-06-09 21:39:37 +02:00 |
cuda_setup
|
Merge branch 'main' into fix/libcuda-to-torch
|
2023-07-06 15:43:42 +09:00 |
nn
|
Only rearrange weight if it exists
|
2023-06-14 19:27:13 +02:00 |
optim
|
Added PagedLion and bf16 Lion.
|
2023-05-23 19:37:38 -07:00 |
research
|
Added is_available_triton guard.
|
2023-04-12 12:10:34 -07:00 |
triton
|
Added triton utils.
|
2023-04-12 12:57:46 -07:00 |
functional.py
|
Added PagedLion and bf16 Lion.
|
2023-05-23 19:37:38 -07:00 |