Commit Graph
4 Commits
Author SHA1 Message Date
chenyuandGitHub 792a494eb8 fix various examples (#4691)
* fix examples that used ax1 and ax2 for transpose

* fix that

* update those
2024-05-22 20:43:21 -04:00
Yixiang GaoandGitHub 094d3d71be with Tensor.train() (#1935)
* add with.train

* remove the rest TODOs

* fix pyflake

* fix pyflake error

* fix mypy
2023-09-28 18:02:31 -07:00
George HotzandGitHub 718ced296c move state to nn/state (#1619) 2023-08-22 07:36:24 -07:00
geohot 87d21ea979 examples: simple conv bn 2023-07-04 13:50:26 -07:00