..
evals
Add MMLU eval to github
2024-05-20 10:20:53 -07:00
activations.h
Add Adam optimizer.
2024-06-06 18:41:36 +00:00
benchmark.cc
Major duplicated code reduction in test/benchmarks
2024-06-14 00:16:25 -07:00
benchmark_helper.cc
Fix debug_prompt and other binaries (internal init)
2024-06-18 06:48:59 -07:00
benchmark_helper.h
Fix Py binding/run_example: use GemmaEnv
2024-06-18 03:20:22 -07:00
benchmarks.cc
Major duplicated code reduction in test/benchmarks
2024-06-14 00:16:25 -07:00
common.cc
Refactor CompressedWeights.
2024-06-17 02:54:54 -07:00
common.h
Internal change.
2024-06-14 06:53:41 -07:00
compress_weights.cc
Further simplification to ForEachTensor, thanks I.K.
2024-06-17 07:12:26 -07:00
configs.h
Move kGriffinLayers into ConfigNoSSM, set kGemmaLayers directly
2024-06-18 07:52:52 -07:00
cross_entropy.cc
Major duplicated code reduction in test/benchmarks
2024-06-14 00:16:25 -07:00
cross_entropy.h
Support all weight types in a single binary.
2024-06-07 09:04:45 -07:00
gemma.cc
Simplify Attention.
2024-06-18 05:08:12 -07:00
gemma.h
Major duplicated code reduction in test/benchmarks
2024-06-14 00:16:25 -07:00
gemma_test.cc
Major duplicated code reduction in test/benchmarks
2024-06-14 00:16:25 -07:00
ops.h
1.15x 7b sfp prefill speedup: Matmul in attention
2024-06-18 01:00:51 -07:00
ops_test.cc
Fix DASSERT - TiledBatch requires at least 2 vectors.
2024-06-17 04:29:01 -07:00
run.cc
Major duplicated code reduction in test/benchmarks
2024-06-14 00:16:25 -07:00
run_mmlu.cc
Major duplicated code reduction in test/benchmarks
2024-06-14 00:16:25 -07:00
weights.cc
Further simplification to ForEachTensor, thanks I.K.
2024-06-17 07:12:26 -07:00
weights.h
Further simplification to ForEachTensor, thanks I.K.
2024-06-17 07:12:26 -07:00
weights_raw.h
Move raw_weights into separate header, used mainly by compress_weights.
2024-06-17 06:17:02 -07:00