13,471 commits

This branch

This branch

All branches

Author	SHA1	Message	Date
George Hotz	4e1a0de392	fix rsub	2020-12-08 10:05:21 -08:00
George Hotz	c4540f1b8c	Support scalars by kartik4949	2020-12-08 09:52:07 -08:00
George Hotz	97fd9c1237	zero_grad there to match readme	2020-12-07 23:12:18 -08:00
George Hotz	c63f950348	need zero grad now	2020-12-07 23:10:43 -08:00
George Hotz	b355cd2571	Mean axis (doesn't work) (#154) ... * mean axis * fixed	2020-12-07 22:58:34 -08:00
George Hotz	38f97c8c80	prepare for ops_ane	2020-12-07 21:54:22 -08:00
George Hotz	7f249ec76d	touch up	2020-12-07 21:51:32 -08:00
Marcel Bischoff	58ccebd7cd	Sum with axis (#153) ... * sum with axis and tests * broken * works again * clean up * Update test_ops.py	2020-12-07 21:49:18 -08:00
George Hotz	ac9fecb05d	lots of notes	2020-12-07 21:40:31 -08:00
George Hotz	8d1500f497	conv neuron	2020-12-07 21:12:52 -08:00
George Hotz	e4bb53b0e9	work out more	2020-12-07 20:32:50 -08:00
George Hotz	4927ad1897	float16 weights in min.weights	2020-12-07 20:15:15 -08:00
George Hotz	3aac9aefce	fix GPU profiling	2020-12-07 20:03:28 -08:00
James Roberts	b2eca6d45f	Format debug output (#152)	2020-12-07 14:07:14 -08:00
George Hotz	c7973cb0a1	ugh buffer_np is bad	2020-12-07 08:07:00 -08:00
George Hotz	088f280dc3	touchups	2020-12-07 07:50:27 -08:00
George Hotz	0cf21881b7	hwx parse w/o macho mods	2020-12-06 23:13:28 -08:00
Josh Smith	aa4161f63e	use classmethods for Tensor helper funcs (#146)	2020-12-06 22:35:43 -08:00
George Hotz	e75a6d1b4b	quadconv	2020-12-06 20:39:50 -08:00
George Hotz	23664c99bd	double conv	2020-12-06 20:26:02 -08:00
George Hotz	1a0f826dc6	highlight the commands	2020-12-06 20:03:21 -08:00
George Hotz	2f1f006003	we have docs	2020-12-06 19:54:03 -08:00
George Hotz	3531e81f0e	dumping ANE docs	2020-12-06 18:58:36 -08:00
George Hotz	6e793e96c3	deeebug	2020-12-06 17:49:17 -08:00
George Hotz	dced0cb44b	oops, path to weights	2020-12-06 16:33:42 -08:00
George Hotz	c57dc61ea7	simple op examples	2020-12-06 16:32:26 -08:00
George Hotz	7babf38617	found concat	2020-12-06 16:27:12 -08:00
George Hotz	ddd6778423	add neuron	2020-12-06 16:24:42 -08:00
George Hotz	f2f2d6aea3	docs and noop	2020-12-06 16:10:44 -08:00
George Hotz	6ba25834ee	found some plists	2020-12-06 15:57:28 -08:00
George Hotz	e2184c20ad	min weights, update golden	2020-12-06 15:29:15 -08:00
George Hotz	d4d8bd0337	make minimal plist for compare	2020-12-06 15:10:15 -08:00
George Hotz	0845ec43c6	compile takes in plist	2020-12-06 14:51:33 -08:00
George Hotz	00312b8ad1	batchnorm work	2020-12-06 14:40:07 -08:00
George Hotz	da514c2918	fix enet init	2020-12-06 13:52:07 -08:00
George Hotz	3b982f2f7a	get_parameters	2020-12-06 13:47:28 -08:00
George Hotz	102e6356e9	replace layer_init_uniform with .uniform	2020-12-06 13:44:31 -08:00
George Hotz	c71a8ef222	remove unused Div op	2020-12-06 13:02:12 -08:00
George Hotz	20f95de408	less lines	2020-12-06 12:56:03 -08:00
George Hotz	629efb391f	lose a few lines	2020-12-06 12:42:41 -08:00
George Hotz	521098cc2f	se optional, track time better	2020-12-06 12:29:42 -08:00
George Hotz	609d11e699	trainer works with CIFAR	2020-12-06 12:20:14 -08:00
George Hotz	80a9c777ba	requires grad, optim in train enet	2020-12-06 11:10:30 -08:00
George Hotz	c66c27d22e	get parameters	2020-12-06 10:45:04 -08:00
George Hotz	51daaa43d4	fix memory leaks, add gc test	2020-12-06 10:34:40 -08:00
George Hotz	1717daa859	reshape doesn't copy anymore	2020-12-06 09:51:09 -08:00
George Hotz	62ee47fef8	add GPUBuffer class	2020-12-06 09:45:13 -08:00
George Hotz	17659f7dd7	gpu speedup, tests work on M1	2020-12-06 09:05:49 -08:00
George Hotz	b8deb36e56	train BS=16 for 32 steps	2020-12-04 10:00:32 -08:00
George Hotz	ad1b225722	oops, i broke fill	2020-12-04 09:53:38 -08:00