LICENSE
README.md
pyproject.toml
trinix/__init__.py
trinix/version.py
trinix.egg-info/PKG-INFO
trinix.egg-info/SOURCES.txt
trinix.egg-info/dependency_links.txt
trinix.egg-info/requires.txt
trinix.egg-info/top_level.txt
trinix/kernels/__init__.py
trinix/kernels/adam_kernel.py
trinix/kernels/adamw_kernel.py
trinix/kernels/alibi_kernel.py
trinix/kernels/attention_kernel.py
trinix/kernels/geglu_kernel.py
trinix/kernels/layernorm_kernel.py
trinix/kernels/lion_kernel.py
trinix/kernels/mish_kernel.py
trinix/kernels/muon_kernel.py
trinix/kernels/quickgelu_kernel.py
trinix/kernels/reglu_kernel.py
trinix/kernels/rmsnorm_kernel.py
trinix/kernels/rope_kernel.py
trinix/kernels/squared_relu_kernel.py
trinix/kernels/swigelu_kernel.py
trinix/kernels/utils.py
trinix/layers/__init__.py
trinix/layers/activation/__init__.py
trinix/layers/activation/geglu.py
trinix/layers/activation/mish.py
trinix/layers/activation/quickgelu.py
trinix/layers/activation/reglu.py
trinix/layers/activation/squared_relu.py
trinix/layers/activation/swigelu.py
trinix/layers/attention/__init__.py
trinix/layers/attention/base.py
trinix/layers/attention/functional.py
trinix/layers/attention/grouped_query_attention.py
trinix/layers/attention/multi_query_attention.py
trinix/layers/attention/multihead_attention.py
trinix/layers/attention/multihead_latent_attention.py
trinix/layers/attention/multihead_self_attention.py
trinix/layers/embeddings/__init__.py
trinix/layers/embeddings/alibi.py
trinix/layers/embeddings/rope.py
trinix/layers/norm/__init__.py
trinix/layers/norm/layernorm.py
trinix/layers/norm/rmsnorm.py
trinix/optim/__init__.py
trinix/optim/adam.py
trinix/optim/adamw.py
trinix/optim/lion.py
trinix/optim/muon.py