Sofie Van Landeghem 43078c347e clean up original tokenizing_distributed_data_loader (#478)

2026-01-31 19:44:12 -08:00

..

__init__.py

initial commit

2025-10-13 06:49:24 -07:00

checkpoint_manager.py

minor helpful message

2026-01-15 03:20:21 +00:00

common.py

i think this is the new torch 2.9+ API for declaring tf32 preference

2026-01-30 17:03:15 +00:00

core_eval.py

initial commit

2025-10-13 06:49:24 -07:00

dataloader.py

clean up original tokenizing_distributed_data_loader (#478)

2026-01-31 19:44:12 -08:00

dataset.py

initial commit

2025-10-13 06:49:24 -07:00

engine.py

fix: pass device_type to compute_init in engine.__main__ (#451)

2026-01-19 17:19:51 -08:00

execution.py

nit delete redundant catch/raise in execute

2025-10-29 08:10:03 -07:00

flash_attention.py

Add Blackwell (SM100) GPU support via SDPA fallback (#475)

2026-01-31 19:42:58 -08:00

gpt.py

Fix generate() crash when top_k=0 (#467)

2026-01-30 09:21:02 -08:00

logo.svg

initial commit

2025-10-13 06:49:24 -07:00

loss_eval.py

fix typos

2025-11-14 11:20:25 +01:00

optim.py

Combine AdamW and Muon into single MuonAdamW optimizer, cleaner, ty @chrisjmccormick for idea/help

2026-01-29 00:52:08 +00:00

report.py

fix small bug where this would break if git stage has deleted files

2026-01-04 19:11:43 +00:00

tokenizer.py

adjust the comment on the regex pattern per recent experimnet see dev/LOG.md

2026-01-13 17:50:39 +00:00

ui.html

Fix conversation scroll to bottom on some browsers + remove duplicated padding (#348)

2025-12-31 13:03:22 -08:00