Ecosyste.ms: Packages

An open API service providing package, version and dependency metadata of many open source software ecosystems and registries.

pypi.org : transformer-lens : 1.11.0

An implementation of transformers tailored for mechanistic interpretability.

Registry - Documentation - Download - JSON
Integrity: sha256-07b2300f2835f888e636... -
purl: pkg:pypi/[email protected]
Published:
Indexed:
Related tag: v1.11.0 - compare

    Loading...
    Readme
    Loading...
    Dependencies
    • nvidia-cublas-cu12 >=12.1.3.1 platform_system == "Linux" and platform_machine == "x86_64" (optional)
    • nvidia-cuda-cupti-cu12 >=12.1.105 platform_system == "Linux" and platform_machine == "x86_64" (optional)
    • nvidia-cuda-nvrtc-cu12 >=12.1.105 platform_system == "Linux" and platform_machine == "x86_64" (optional)
    • nvidia-cuda-runtime-cu12 >=12.1.105 platform_system == "Linux" and platform_machine == "x86_64" (optional)
    • nvidia-cudnn-cu12 >=8.9.2.26 platform_system == "Linux" and platform_machine == "x86_64" (optional)
    • nvidia-cufft-cu12 >=11.0.2.54 platform_system == "Linux" and platform_machine == "x86_64" (optional)
    • nvidia-curand-cu12 >=10.3.2.106 platform_system == "Linux" and platform_machine == "x86_64" (optional)
    • nvidia-cusolver-cu12 >=11.4.5.107 platform_system == "Linux" and platform_machine == "x86_64" (optional)
    • nvidia-cusparse-cu12 >=12.1.0.106 platform_system == "Linux" and platform_machine == "x86_64" (optional)
    • nvidia-nccl-cu12 >=2.18.1 platform_system == "Linux" and platform_machine == "x86_64" (optional)
    • nvidia-nvtx-cu12 >=12.1.105 platform_system == "Linux" and platform_machine == "x86_64" (optional)
    • triton >=2.1.0 platform_system == "Linux" and platform_machine == "x86_64" (optional)
    • numpy >=1.26 python_version >= "3.12" and python_version < "3.13" (optional)
    • numpy >=1.20,<1.25 python_version >= "3.8" and python_version < "3.9" (optional)
    • numpy >=1.24 python_version >= "3.9" and python_version < "3.12" (optional)
    • accelerate >=0.23.0
    • beartype >=0.14.1,<0.15.0
    • datasets >=2.7.1
    • einops >=0.6.0
    • fancy-einsum >=0.0.3
    • jaxtyping >=0.2.11
    • pandas >=1.1.5
    • rich >=12.6.0
    • torch >=1.10,!=2.0,!=2.1.0
    • tqdm >=4.64.1
    • transformers >=4.25.1
    • typing-extensions *
    • wandb >=0.13.5