Code

merge torchviz+latex into main#12

Code

Merged

plutonium-239 merged 13 commits intomainplutonium-239/memsave_torch:mainfrom torchviz+latexplutonium-239/memsave_torch:torchviz+latexCopy head branch name to clipboardAug 22, 2024

+381

Commits

Commits on Aug 22, 2024

• visualize the backward graph of specific blocks

  Show description for 3862933
  plutonium-239

  committed
  3862933

• minor

  Show description for 9d94bef
  plutonium-239

  committed
  9d94bef

• fixes for vis

  Show description for b2ae54f
  plutonium-239

  committed
  b2ae54f

• more vis fixes

  Show description for 8e98587
  plutonium-239

  committed
  8e98587

• rms layer norm

  Show description for 14f586f
  plutonium-239

  committed
  14f586f

• code for making latex table

  Show description for 2d997c6
  plutonium-239

  committed
  2d997c6

• new implementation of dropout

  Show description for 2c0ccba
  plutonium-239

  committed
  2c0ccba

• adding llama3 and phi3

  Show description for 6e6fb00
  plutonium-239

  committed
  6e6fb00

• add VLM architecture in estimate, refactor cases to refer to a central location

  Show description for 4568c03
  plutonium-239

  committed
  4568c03

• torchviz visualize elements

  Show description for 6c26608
  plutonium-239

  committed
  6c26608

• format + prep for merge

  Show description for bdf34af
  plutonium-239

  committed
  bdf34af

• minor improvements to latex table script and batch norm in eval

  Show description for 8eed547
  plutonium-239

  committed
  8eed547

• minor updates:

  Show description for 22957d9
  plutonium-239

  committed
  22957d9