tinygrad/tinygrad
chenyu 7afca52796
replace pow in LAMB by tracking b1**t and b2**t per step (#4582)
* replace pow in LAMB by tracking b1**t and b2**t per step

* remove t, add [self.b1_t, self.b2_t] to return

* adam has one less kernel
2024-05-14 13:08:22 -04:00
..
codegen all uops are now cachable (#4564) 2024-05-12 22:34:35 -07:00
engine small scheduler refactor (#4569) 2024-05-13 10:47:39 +03:00
features move scheduleitem to schedule.py (#4541) 2024-05-11 21:13:04 -07:00
nn replace pow in LAMB by tracking b1**t and b2**t per step (#4582) 2024-05-14 13:08:22 -04:00
renderer Optimize PTX gated loads index calculation (#4304) 2024-05-13 10:14:01 -07:00
runtime remove rhip (#4579) 2024-05-14 17:58:19 +03:00
shape add consecutive property to shapetracker 2024-05-02 10:41:28 -07:00
__init__.py add mnist tutorial 2024-04-25 16:08:32 +08:00
device.py all realize 2 (#4527) 2024-05-10 22:43:09 -07:00
dtype.py move acc_dt out of lazy (#4382) 2024-05-06 07:41:25 -07:00
function.py move acc_dt out of lazy (#4382) 2024-05-06 07:41:25 -07:00
helpers.py add cpu objdump to LLVM/CLANG (#4537) 2024-05-11 14:28:44 -07:00
lazy.py bring buffer back to device (#4517) 2024-05-10 11:22:31 -07:00
ops.py move scheduleitem to schedule.py (#4541) 2024-05-11 21:13:04 -07:00
tensor.py setitem in-place operator tests (#4577) 2024-05-14 01:28:02 -04:00