| .. |
|
fused_moe
|
41beab5dc1
add exllamav2 tensor paralell, fused MoE for GPTQ/AWQ
|
9 months ago |
|
ops
|
9181fa0396
feat: Triton kernels for sampling (#383)
|
9 months ago |
|
quantization
|
bd0ddf1cfe
feat: EETQ quantization (#408)
|
9 months ago |
|
__init__.py
|
07aa2a492f
upstream: add option to specify tokenizer
|
1 year ago |
|
activation.py
|
50c2434267
move megatron to a top-level directory
|
9 months ago |
|
layernorm.py
|
e31c6f0b45
feat: refactor modeling logic and support more models (#274)
|
11 months ago |
|
linear.py
|
50c2434267
move megatron to a top-level directory
|
9 months ago |
|
logits_processor.py
|
50c2434267
move megatron to a top-level directory
|
9 months ago |
|
rejection.py
|
f8dfac6372
chore: attention refactor and upstream sync apr01 (#365)
|
9 months ago |
|
rotary_embedding.py
|
c8a91b0b96
rope: get_device() -> device
|
9 months ago |
|
sampler.py
|
fa083286e3
Speculative Decoding Part 4: Lookahead scheduling (#402)
|
9 months ago |
|
vocab_parallel_embedding.py
|
50c2434267
move megatron to a top-level directory
|
9 months ago |
AlpinDale
bd0ddf1cfe
feat: EETQ quantization (#408)