Ecosyste.ms: Repos

An open API service providing repository metadata for many open source software ecosystems.

GitHub / alan-cooney / TransformerLens

Fork of neelnanda-io/TransformerLens
Stars: 0
Forks: 0
Open Issues: 0

License: mit
Language: Python
Repo Size: 22.6 MB
Dependencies: 256

Created: over 1 year ago
Updated: 7 months ago
Last pushed: 7 months ago
Last synced: 7 months ago

Readme

.github/workflows/checks.yml actions

.github/workflows/gh-pages.yml actions

.github/workflows/release.yml actions

.devcontainer/Dockerfile docker

pyproject.toml pypi

accelerate >=0.23.0
beartype ^0.14.1
datasets >=2.7.1
einops >=0.6.0
fancy-einsum >=0.0.3
jaxtyping >=0.2.11
numpy --- - !ruby/hash:ActiveSupport::HashWithIndifferentAccess version: ">=1.21" python: "<3.10" - !ruby/hash:ActiveSupport::HashWithIndifferentAccess version: ">=1.23" python: ">=3.10"
nvidia-cublas-cu12 ==12.1.3.1
nvidia-cuda-cupti-cu12 ==12.1.105
nvidia-cuda-nvrtc-cu12 ==12.1.105
nvidia-cuda-runtime-cu12 ==12.1.105
nvidia-cudnn-cu12 ==8.9.2.26
nvidia-cufft-cu12 ==11.0.2.54
nvidia-curand-cu12 ==10.3.2.106
nvidia-cusolver-cu12 ==11.4.5.107
nvidia-cusparse-cu12 ==12.1.0.106
nvidia-nccl-cu12 ==2.18.1
nvidia-nvtx-cu12 ==12.1.105
pandas >=1.1.5
python >=3.7.4,<4.0
rich >=12.6.0
torch --- - !ruby/hash:ActiveSupport::HashWithIndifferentAccess version: ">=1.10, <2.0.0" python: "<3.8" - !ruby/hash:ActiveSupport::HashWithIndifferentAccess version: ">=1.10" python: ">=3.8"
tqdm >=4.64.1
transformers >=4.25.1
triton ==2.1.0
typing-extensions *
wandb >=0.13.5

poetry.lock pypi