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

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

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

use comgr to compile (#3248)

2024-01-26 18:27:49 -08:00

junk

coder.py can write and run code (#2439)

2023-11-25 12:27:54 -08:00

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

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

augment.py

[ready] Replacing os with pathlib (#1708)

2023-08-30 10:41:08 -07:00

disk_read_speed.py

io_uring for copies from disk (#5035)

2024-06-21 11:36:51 +03:00

dump_cache.py

wow how did i think that was okay (#2339)

2023-11-16 21:21:11 -08:00

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

Fix: Jacobian tests [WIP] (#1126)

2023-07-05 15:36:22 -07:00

hip_events.py

move autogen to runtime/autogen (#3254)

2024-01-26 12:44:19 -08:00

introspection.py

bring buffer back to device (#4517)

2024-05-10 11:22:31 -07:00

lr_scheduler.py

use at least float32 for optim.lr (#4297)

2024-04-25 14:42:28 -04:00

mcts_search.py

AST is UOp (#6030)

2024-08-16 22:09:00 +03:00

multitensor.py

multitensor start (#2676)

2023-12-07 17:07:05 -08:00

onnx_ops.py

tiny import cleanup and fix typo (#6692)

2024-09-23 21:48:23 -04:00

onnx.py

fix buffer shape regression from onnx (#6678)

2024-09-23 16:58:42 +08:00

ops.py

add void type to uop (#6471)

2024-09-11 18:16:28 +08:00

ring_copy.py

ring copy example (#3185)

2024-01-19 23:34:30 -05:00

thneed.py

new style device (#2530)

2023-11-30 17:07:16 -08:00

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

hotfix: copy size is in bytes

2024-01-17 16:44:15 +00:00