AlpinDale 9d81716bfd [v0.5.3] Release Candidate (#388) 8 months ago
..
fused_moe 9d81716bfd [v0.5.3] Release Candidate (#388) 8 months ago
ops 9181fa0396 feat: Triton kernels for sampling (#383) 9 months ago
__init__.py 07aa2a492f upstream: add option to specify tokenizer 1 year ago
activation.py 9d81716bfd [v0.5.3] Release Candidate (#388) 8 months ago
layernorm.py e31c6f0b45 feat: refactor modeling logic and support more models (#274) 11 months ago
linear.py 9d81716bfd [v0.5.3] Release Candidate (#388) 8 months ago
logits_processor.py 9d81716bfd [v0.5.3] Release Candidate (#388) 8 months ago
rejection.py 9d81716bfd [v0.5.3] Release Candidate (#388) 8 months ago
rotary_embedding.py 9d81716bfd [v0.5.3] Release Candidate (#388) 8 months ago
sampler.py 9d81716bfd [v0.5.3] Release Candidate (#388) 8 months ago
vocab_parallel_embedding.py 9d81716bfd [v0.5.3] Release Candidate (#388) 8 months ago