env.yaml

name: llmtt
channels:
  - pytorch
  - nvidia
  - conda-forge
dependencies:
  - python
  - pytorch
  - pytorch-cuda=11.8
  - nodejs
  - yarn
  - pip
  - pip:
      - datasets
      - einops
      - fancy_einsum
      - jaxtyping
      - networkx
      - plotly
      - pyinstrument
      - setuptools
      - streamlit
      - streamlit_extras
      - tokenizers
      - transformer_lens
      - transformers
      - pytest  # fixes wrong dependencies of transformer_lens