tinygrad/extra
samm393 19c11792fd
Flux.1 (#6334)
* initial commit

* whitespace

* get rid of torch import

* indentation

* less hardcoding

* add flux.1-dev

* jit

* no double

* t5 tidy up

* validation image

* reuse sdxl autoencoder

* typing changes

* empty lines

* remove unneeded comments

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
2024-09-24 10:08:04 +08:00
..
accel
assembly lowerer is kernel [run_process_replay] (#5437) 2024-07-12 18:50:55 -07:00
backends UOps.PHI -> UOps.ASSIGN [run_process_replay] (#6383) 2024-09-06 12:38:35 +08:00
datasets UOps.VALID try 2 (#6623) 2024-09-21 14:19:25 +08:00
disassemblers/adreno move disassemblers and openpilot (#4592) 2024-05-14 19:30:02 -07:00
dsp add qcom dsp runtime (#6112) 2024-09-13 21:01:33 +03:00
gemm add qcom dsp runtime (#6112) 2024-09-13 21:01:33 +03:00
hip_gpu_driver feat: autogen from kernel register offset headers (#6056) 2024-08-12 14:08:35 -07:00
hiprtc
junk
mockgpu autogen cleanup (#6064) 2024-08-14 20:20:35 +03:00
models Flux.1 (#6334) 2024-09-24 10:08:04 +08:00
nv_gpu_driver update pylint path to check indent/space for all (#6022) 2024-08-10 14:41:09 -04:00
optimization viz more work [run_process_replay] (#6568) 2024-09-17 19:27:09 +08:00
qcom_gpu_driver qcom updated ioctl (#6627) 2024-09-20 18:02:40 +08:00
resnet18 beat mlx at resnet 18 (#6611) 2024-09-20 11:28:01 +08:00
archprobe.py
augment.py
disk_read_speed.py io_uring for copies from disk (#5035) 2024-06-21 11:36:51 +03:00
dump_cache.py
export_model.py all realize 2 (#4527) 2024-05-10 22:43:09 -07:00
f16_w_uint32.py fix various examples (#4691) 2024-05-22 20:43:21 -04:00
gradcheck.py
hip_events.py
introspection.py bring buffer back to device (#4517) 2024-05-10 11:22:31 -07:00
lr_scheduler.py
mcts_search.py AST is UOp (#6030) 2024-08-16 22:09:00 +03:00
multitensor.py
onnx.py fix buffer shape regression from onnx (#6678) 2024-09-23 16:58:42 +08:00
onnx_ops.py tiny import cleanup and fix typo (#6692) 2024-09-23 21:48:23 -04:00
ops.py add void type to uop (#6471) 2024-09-11 18:16:28 +08:00
ring_copy.py
thneed.py
threefry.py feat: example and extra tweaks (#6310) 2024-08-28 19:26:11 -07:00
to_movement_ops.py update CI tests in extra with UOp AST (#6290) 2024-08-28 22:26:50 +03:00
training.py tinytqdm.set_description and tinytrange (#5101) 2024-06-22 14:45:06 -04:00
transfer_speed.py