Repository navigation

Arch2 - #105

Open

ClashLuke wants to merge 24 commits into

mainfrom

Commits on Dec 30, 2022

test(optimizer): no truegrad
ClashLuke
committed
00c269c

Commits on Jan 22, 2023

feat(model): use mish + merge mixer/conv
ClashLuke
committed
cdc8d0d
Revert "test(optimizer): no truegrad"
ClashLuke
committed
ae6b483
fix(conv): remove leak
ClashLuke
committed
bf68bf6
fix: lots of minor issue
ClashLuke
committed
99f2d25
feat(tests): add test script
ClashLuke
committed
e28bd98
style(main): fix type hint indentation
ClashLuke
committed
f5dbac3

Commits on Jan 24, 2023

perf(dense): use X+GLU+Mixer+Shift as input to dense; add more efficient scale_norm_act_conv
ClashLuke
committed
6e7130e

Commits on Jan 25, 2023

style: fix deepsource issues
ClashLuke
committed
0a1ea7c

Commits on Feb 8, 2023

fix(dense): use correct lax.pad dtype
ClashLuke
committed
504153f
fix(dense): max path doesn't work with pad
ClashLuke
committed
a12d843
fix(dense): *features was for numel, not idx in dim
ClashLuke
committed
81b153a
fix: add missing () to with_context
ClashLuke
committed
7907a3d
fix(dense): handle unreshapable mask case
ClashLuke
committed
9292d83
fix(activate): casting issue
ClashLuke
committed
4b8eb08
fix(norm): normalize out_features
ClashLuke
committed
99ddce3
fix(norm): revert
ClashLuke
committed
b6da249
test(norm): show shapes
ClashLuke
committed
ee000be
test(norm): show name cache and prefix
ClashLuke
committed
553fd42
fix(norm): change shape in norm conv
ClashLuke
committed
289a9f1
fix(norm): use correct shape
ClashLuke
authored
2b728d2
fix(norm): only get shape if array
ClashLuke
authored
ea36c42
v4
ClashLuke
authored
d57a657

Commits on Jan 20, 2024

perf(dense): matmul during memory op
ClashLuke
authored
ba2e311