Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

sarus / sarus-llm python

Repository URL to install this package:

Version: 
1.1.3  ▾
Details    
sarus-llm / sarus_llm / models / modules / __init__.py
Size: Mime: 
from .attention import CausalSelfAttention
from .feed_forward import FeedForward
from .kv_cache import KVCache
from .position_embeddings import RotaryPositionalEmbeddings
from .rms_norm import RMSNorm
from .transformer import TransformerDecoder, TransformerDecoderLayer

__all__ = [
    "CausalSelfAttention",
    "FeedForward",
    "KVCache",
    "RotaryPositionalEmbeddings",
    "RMSNorm",
    "TransformerDecoder",
    "TransformerDecoderLayer",
]

Products

About

Resources

Contact Gemfury