AQIT 0.1.0
Loading...
Searching...
No Matches
aquin.compute.sae.SparseAutoencoder Member List

This is the complete list of members for aquin.compute.sae.SparseAutoencoder, including all inherited members.

__init__(self, int d_model=2048, int n_features=16384)aquin.compute.sae.SparseAutoencoder
_coerce_checkpoint(cls, Any ckpt)aquin.compute.sae.SparseAutoencoderprotected
_find_tensor(cls, dict tensors, *str names)aquin.compute.sae.SparseAutoencoderprotected
_infer_dims(cls, dict tensors)aquin.compute.sae.SparseAutoencoderprotected
_load_safetensors(cls, Path path, str device="cuda")aquin.compute.sae.SparseAutoencoderprotected
_meta_int(cls, dict[str, Any] meta, *str keys)aquin.compute.sae.SparseAutoencoderprotected
_normalise_decoder(self)aquin.compute.sae.SparseAutoencoderprotected
_remap_tensors(cls, dict[str, torch.Tensor] tensors, "SparseAutoencoder" sae)aquin.compute.sae.SparseAutoencoderprotected
b_decaquin.compute.sae.SparseAutoencoder
b_encaquin.compute.sae.SparseAutoencoder
b_preaquin.compute.sae.SparseAutoencoder
d_modelaquin.compute.sae.SparseAutoencoder
decode(self, torch.Tensor f)aquin.compute.sae.SparseAutoencoder
encode(self, torch.Tensor x)aquin.compute.sae.SparseAutoencoder
forward(self, torch.Tensor x)aquin.compute.sae.SparseAutoencoder
get_top_features(self, torch.Tensor x, int k=10)aquin.compute.sae.SparseAutoencoder
load(cls, str|Path path, str device="cuda")aquin.compute.sae.SparseAutoencoder
n_featuresaquin.compute.sae.SparseAutoencoder
save(self, str|Path path)aquin.compute.sae.SparseAutoencoder
W_decaquin.compute.sae.SparseAutoencoder
W_encaquin.compute.sae.SparseAutoencoder