15 commits

Author	SHA1	Message	Date
George Hotz	e4d8e03954	more test	2026-03-16 12:37:16 +08:00
George Hotz	09574a096a	maketuple	2026-03-16 12:04:39 +08:00
George Hotz	b89c233917	fix	2026-03-16 11:56:36 +08:00
George Hotz	00d847afdf	single backward gradient	2026-03-16 11:44:15 +08:00
George Hotz	8d75eed0a4	fix precompile	2026-03-16 11:33:52 +08:00
George Hotz	e59cbf78bd	Add TUPLE and GETTUPLE	2026-03-16 11:22:03 +08:00
chenyu	14d1c5fdfd	assign fusion tests on detach and contiguous_backward (#15092)	2026-03-02 15:21:51 -05:00
George Hotz	bb84e389cf	functions for llama trainer (#15045) ... * functions for llama trainer * function there * axis match * fix multi * lil cleaner * there's a bug with HK_FLASH_ATTENTION * training functions * for commit	2026-02-28 12:15:18 +08:00
George Hotz	010d2790ce	fix multi minimal (#15044)	2026-02-27 14:31:58 +08:00
George Hotz	fe3ee8c27e	fix symbolic shapes in calls (#15021) ... * fix symbolic shapes in calls * fix after in the big graph * real tests	2026-02-26 17:17:18 +08:00
George Hotz	2655655a0c	call gradient creates a call (#15020) ... * function creates a full subgraph * tests * fix var * fix tests * implict assign/contig * move kv init	2026-02-26 14:15:29 +08:00
chenyu	ed9d475a12	assign tests with test_function (#15015)	2026-02-25 16:15:59 -05:00
George Hotz	0d35b67f2c	revert realize to only be buffers (#15008) ... * revert realize to only be buffers * fix that * broken attention * Revert "broken attention" This reverts commit a23c3cd96c. * and that	2026-02-25 22:43:06 +08:00
George Hotz	68831cd852	add more tests to test_function (#15003) ... * add more tests to test_function * add function to llm * function decorator on llm * works * symbolic fixups * minimum change * implicit inputs * don't actually update llama yet	2026-02-25 18:42:06 +08:00
George Hotz	e3fa9896b7	start function and add walk rewrite (#14992) ... * start function and add walk rewrite * work * add function on feed_forward * llm progress * stuff * none of that	2026-02-25 13:56:27 +08:00