Tinygrad v0 | 0F87A42B18D43BDEF7300EE909544939916E14A3

Tinygrad v0

Name	tinygrad-v0.9.0	DOWNLOAD Copy Link
Total Size	11.7 MB
Total Files	499
Last Seen	2024-10-06 01:40
Hash	0F87A42B18D43BDEF7300EE909544939916E14A3

/
autogen_stubs.sh	8.2 KB
LICENSE	1.1 KB
mkdocs.yml	3.2 KB
mypy.ini	0.3 KB
README.md	8.0 KB
ruff.toml	0.8 KB
serve_docs.sh	0.0 KB
setup.py	2.3 KB
sz.py	3.6 KB
/docs/
abstractions2.py	3.9 KB
abstractions3.py	1.7 KB
CNAME	0.0 KB
developer.md	2.3 KB
dtypes.md	0.2 KB
env_vars.md	2.3 KB
favicon.svg	0.8 KB
function.md	0.7 KB
index.md	2.7 KB
logo_tiny_dark.svg	0.5 KB
logo_tiny_light.svg	0.5 KB
mnist.md	6.1 KB
nn.md	0.7 KB
quickstart.md	12.9 KB
showcase.md	1.6 KB
/docs/showcase/
mnist_by_tinygrad.jpg	15.1 KB
stable_diffusion_by_tinygrad.jpg	89.5 KB
yolo_by_tinygrad.jpg	133.9 KB
yolov8_showcase_image.png	1.1 MB
/docs/tensor/
creation.md	0.6 KB
index.md	0.7 KB
movement.md	0.6 KB
ops.md	2.6 KB
/examples/
beautiful_cartpole.py	5.1 KB
beautiful_mnist.py	2.0 KB
beautiful_mnist_multigpu.py	2.3 KB
coder.py	4.5 KB
compile_efficientnet.py	3.0 KB
compile_tensorflow.py	3.5 KB
conversation.py	15.2 KB
efficientnet.py	2.9 KB
gpt2.py	9.4 KB
handcode_resnet50_opt.py	3.3 KB
hlb_cifar10.py	19.4 KB
index.html	4.1 KB
llama.py	22.5 KB
llama3.py	13.3 KB
mamba.py	13.5 KB
mask_rcnn.py	9.7 KB
mixtral.py	3.2 KB
mnist_gan.py	4.0 KB
serious_mnist.py	4.6 KB
simple_conv_bn.py	0.5 KB
so_vits_svc.py	40.2 KB
stable_diffusion.py	24.0 KB
stable_diffusion_seed0.png	490.2 KB
train_efficientnet.py	3.1 KB
train_resnet.py	1.4 KB
transformer.py	1.6 KB
vgg7.py	9.8 KB
vit.py	1.5 KB
vits.py	54.1 KB
whisper.py	18.2 KB
yolov3.py	16.9 KB
yolov8-onnx.py	0.6 KB
yolov8.py	19.3 KB
/examples/conversation_data/
pre_prompt_gary.yaml	0.7 KB
pre_prompt_george.yaml	1.2 KB
pre_prompt_lexie.yaml	0.8 KB
pre_prompt_stacy.yaml	0.8 KB
/examples/llm.c/
export.py	4.7 KB
train_gpt2.py	6.9 KB
/examples/llm.c/ubench/
matmul.c	2.2 KB
/examples/mlperf/
dataloader.py	9.3 KB
helpers.py	13.7 KB
initializers.py	4.5 KB
losses.py	0.3 KB
lr_schedulers.py	1.2 KB
metrics.py	2.0 KB
model_eval.py	9.7 KB
model_spec.py	2.0 KB
model_train.py	28.8 KB
README	0.5 KB
/.../training_submission_v4.0/tinycorp/benchmarks/resnet/implementations/tinybox_green/
dev_beam.sh	0.4 KB
dev_run.sh	0.4 KB
README.md	1.2 KB
run_and_time.sh	0.7 KB
/.../training_submission_v4.0/tinycorp/benchmarks/resnet/implementations/tinybox_red/
dev_beam.sh	0.4 KB
dev_run.sh	0.4 KB
README.md	1.2 KB
run_and_time.sh	0.7 KB
setup.sh	0.2 KB
/.../training_submission_v4.0/tinycorp/systems/
tinybox_green.json	1.3 KB
tinybox_red.json	1.3 KB
/examples/openpilot/
compile2.py	8.7 KB
go.sh	0.1 KB
/examples/rl/
lightupbutton.py	1.5 KB
/examples/sovits_helpers/
preprocess.py	11.7 KB
/examples/vgg7_helpers/
waifu2x.py	8.0 KB
/examples/webgl/yolov8/
compile.py	1.1 KB
index.html	8.9 KB
/examples/webgpu/stable_diffusion/
compile.py	9.9 KB
f16_to_f32.js	2.7 KB
index.html	19.5 KB
/extra/accel/ane/1_build/
coreml_ane.py	1.3 KB
run.swift	1.1 KB
test.mlmodel	0.1 KB
/extra/accel/ane/2_compile/
ane.py	6.2 KB
aneregs.json	24.0 KB
compile.m	1.9 KB
compile.mm	3.6 KB
compile.sh	0.2 KB
dcompile.py	0.4 KB
hwx_parse.py	3.8 KB
min.weights	2.0 KB
min_uint8.weights	0.3 KB
model.espresso.weights	0.2 KB
model.hwx.golden	32.8 KB
net.additional.weights	0.0 KB
net.plist	2.8 KB
struct_recover.py	0.7 KB
twos.weights	0.3 KB
/extra/accel/ane/2_compile/debug/
README	0.1 KB
/extra/accel/ane/2_compile/plists/
broadcast.plist	4.6 KB
concat.plist	3.0 KB
gemm.plist	3.2 KB
goc.plist	2.0 KB
inputview.plist	3.8 KB
neuron.plist	1.9 KB
reshape.plist	2.2 KB
scaled.plist	3.2 KB
sum.plist	2.6 KB
/extra/accel/ane/2_compile/simple/
concat.plist	1.7 KB
conv.plist	2.0 KB
convneuron.plist	2.8 KB
convuint8.plist	3.0 KB
doubleconv.plist	3.1 KB
doubleconvrev.plist	3.1 KB
doubleconvsout.plist	2.9 KB
doubleneuron.plist	1.7 KB
gemm.plist	1.9 KB
goc.plist	1.6 KB
neuron.plist	1.3 KB
quadconv.plist	4.4 KB
reshape.plist	1.8 KB
/extra/accel/ane/3_run/
build.sh	0.2 KB
entitlements.xml	0.2 KB
h11ane.h	4.4 KB
test.mm	6.0 KB
/extra/accel/ane/amfi/
new_patch.py	3.2 KB
/extra/accel/ane/
aneregs	8.9 KB
README.md	4.3 KB
README.old	10.6 KB
/extra/accel/ane/lib/
ane.mm	7.0 KB
ane.py	6.2 KB
aneregs.json	24.0 KB
build.sh	0.2 KB
entitlements.xml	0.2 KB
h11ane.h	4.4 KB
sign_python.sh	0.2 KB
testconv.py	2.0 KB
/extra/accel/ane/ops/
concat.hwx	49.2 KB
conv.hwx	32.8 KB
gemm.hwx	557.1 KB
relu.hwx	49.2 KB
sigmoid.hwx	32.8 KB
sum.hwx	49.2 KB
/extra/accel/ane/tinygrad/
ops_ane.py	1.3 KB
/extra/accel/intel/
benchmark_matmul.py	1.4 KB
go.sh	0.1 KB
joint_matrix_bfloat16.cpp	6.3 KB
README	0.1 KB
/extra/accel/
MAPPING	1.5 KB
README	0.2 KB
/extra/accel/tpu/logs/
tpu_driver.t1v-n-852cd0d5-w-0.taylor.log.INFO.20210619-062914.26926.gz	1.4 MB
/extra/accel/tpu/
README.md	4.1 KB
/extra/accel/tpu/tfexample/
libtpu.h	13.0 KB
libtpu_client.c	6.3 KB
/extra/
archprobe.py	5.5 KB
augment.py	1.5 KB
disk_read_speed.py	4.4 KB
dump_cache.py	0.7 KB
export_model.py	17.9 KB
f16_w_uint32.py	1.3 KB
gradcheck.py	1.5 KB
hip_events.py	0.4 KB
introspection.py	2.1 KB
lr_scheduler.py	3.8 KB
multitensor.py	2.0 KB
onnx.py	11.0 KB
onnx_ops.py	43.1 KB
ring_copy.py	0.5 KB
thneed.py	11.6 KB
threefry.py	0.4 KB
to_movement_ops.py	8.2 KB
training.py	2.2 KB
transfer_speed.py	0.3 KB
/extra/assembly/
assembly.py	10.5 KB
assembly_arm64.py	9.4 KB
assembly_ptx.py	6.7 KB
assembly_rdna.py	11.1 KB
/extra/assembly/ptx/
test.py	0.6 KB
/extra/assembly/rocm/kernel_crashes/
dump	2.7 KB
dump2	2.8 KB
dump3	1.5 KB
dump4	3.2 KB
/extra/assembly/rocm/
NOTES	0.6 KB
rocm_clone.sh	0.6 KB
rocm_from_scratch.sh	1.7 KB
rocm_setup.sh	0.4 KB
/extra/assembly/rocm/rdna3/
asm.py	2.6 KB
prog.s	2.5 KB
/extra/assembly/rocm/sniffer/
build.sh	1.0 KB
sniff.cc	13.1 KB
/extra/backends/
cstyle.py	5.2 KB
graph_hip.py	2.1 KB
ops_hip.py	9.5 KB
ops_rhip.py	0.9 KB
ops_webgl.py	2.6 KB
ops_webgpu.py	2.7 KB
triton.py	7.5 KB
/extra/datasets/
__init__.py	2.4 KB
coco.py	5.8 KB
fake_imagenet_from_mnist.py	1.3 KB
imagenet.py	3.1 KB
imagenet_download.py	3.2 KB
kits19.py	10.5 KB
librispeech.py	2.9 KB
openimages.py	9.0 KB
preprocess_imagenet.py	0.7 KB
sops.gz	706.0 KB
squad.py	4.7 KB
wikipedia.py	15.2 KB
wikipedia_download.py	3.0 KB
/extra/disassemblers/adreno/
__init__.py	0.5 KB
disasm-a3xx.c	39.6 KB
instr-a3xx.h	29.3 KB
ir3.h	47.8 KB
README	0.1 KB
shader_enums.h	30.3 KB
/extra/disassemblers/adreno/util/
bitscan.h	7.5 KB
bitset.h	12.1 KB
list.h	9.4 KB
macros.h	10.4 KB
/extra/gemm/
amx.py	7.6 KB
cuda_matmul.py	4.0 KB
fuzz_matmul.py	1.8 KB
gemm.c	4.5 KB
gemm.py	0.6 KB
hip_matmul.py	6.3 KB
intel_xmx.py	1.8 KB
jax_pmatmul.py	0.9 KB
metal_conv.py	1.6 KB
metal_matmul.py	5.6 KB
metal_matvec.py	5.1 KB
mlx_matmul.py	0.2 KB
simple_conv.py	1.3 KB
simple_matmul.py	1.3 KB
simple_matvec.py	1.1 KB
tf_gemm.py	1.1 KB
torch_gemm.py	0.5 KB
tvm_gemm.py	1.5 KB
/.../hip_gpu_driver/
hip_ioctl.py	4.0 KB
nvd.h	16.7 KB
sdma_registers.h	13.3 KB
test_kfd_2.py	12.1 KB
test_pm4.py	6.3 KB
test_sdma_fun.py	2.1 KB
/extra/hiprtc/
hiprtc_runtime.h	563.3 KB
/extra/junk/
sentencepiece_model_pb2.py	6.6 KB
/extra/mockgpu/amd/
amddriver.py	6.1 KB
amdgpu.py	9.6 KB
/extra/mockgpu/
driver.py	3.7 KB
gpu.py	0.2 KB
mockgpu.py	7.7 KB
/extra/mockgpu/nv/
nvdriver.py	11.4 KB
nvgpu.py	8.6 KB
/extra/models/
bert.py	10.4 KB
convnext.py	2.5 KB
efficientnet.py	6.8 KB
llama.py	10.7 KB
mask_rcnn.py	41.7 KB
resnet.py	6.4 KB
retinanet.py	11.4 KB
rnnt.py	8.0 KB
transformer.py	2.8 KB
unet3d.py	2.5 KB
vit.py	4.2 KB
/.../nv_gpu_driver/
clc6c0qmd.h	58.0 KB
nv_ioctl.py	11.1 KB
/extra/optimization/
extract_policynet.py	3.7 KB
extract_sa_pairs.py	4.3 KB
generate_dataset.sh	0.9 KB
get_action_space.py	1.1 KB
helpers.py	3.5 KB
pretrain_valuenet.py	2.7 KB
rl.py	2.6 KB
run_qnet.py	1.2 KB
search.py	1.4 KB
test_beam_search.py	2.4 KB
test_net.py	2.2 KB
test_time_linearizer.py	1.0 KB
/.../qcom_gpu_driver/
adreno_pm4.xml	85.7 KB
gen.sh	0.0 KB
msm_kgsl.h	45.4 KB
msm_kgsl.py	30.5 KB
opencl_ioctl.py	6.3 KB
/test/
Dockerfile	0.3 KB
helpers.py	2.3 KB
test_arange.py	0.5 KB
test_assign.py	12.8 KB
test_const_folding.py	10.4 KB
test_conv.py	4.2 KB
test_conv_shapetracker.py	0.9 KB
test_copy_speed.py	2.5 KB
test_custom_function.py	4.8 KB
test_device_speed.py	1.2 KB
test_dtype.py	34.6 KB
test_dtype_alu.py	8.3 KB
test_fusion_op.py	1.7 KB
test_fuzz_shape_ops.py	3.0 KB
test_gc.py	1.1 KB
test_image_dtype.py	2.8 KB
test_jit.py	10.9 KB
test_kernel_cache.py	0.7 KB
test_lazybuffer.py	4.0 KB
test_lazyop.py	1.1 KB
test_linearizer.py	89.2 KB
test_linearizer_failures.py	70.9 KB
test_linearizer_overflows.py	14.9 KB
test_masked_st.py	0.9 KB
test_method_cache.py	1.7 KB
test_multitensor.py	31.4 KB
test_net_speed.py	2.5 KB
test_nn.py	15.4 KB
test_ops.py	94.8 KB
test_optim.py	6.0 KB
test_pattern_matcher.py	4.4 KB
test_pickle.py	1.8 KB
test_randomness.py	10.2 KB
test_sample.py	0.7 KB
test_schedule.py	27.4 KB
test_search.py	7.4 KB
test_setitem.py	4.6 KB
test_specific_conv.py	2.3 KB
test_speed_v_torch.py	11.6 KB
test_subbuffer.py	1.9 KB
test_symbolic_jit.py	8.5 KB
test_symbolic_ops.py	7.2 KB
test_symbolic_shapetracker.py	7.6 KB
test_tensor.py	21.6 KB
test_tensor_data.py	1.5 KB
test_tensor_variable.py	1.8 KB
test_to_numpy.py	0.4 KB
test_uop_graph.py	3.2 KB
test_uops.py	12.9 KB
test_uops_stats.py	3.1 KB
test_winograd.py	2.5 KB
test_zero_copy.py	0.9 KB
/test/external/
external_benchmark_hip_compile.py	1.2 KB
external_benchmark_load_stable_diffusion.py	0.6 KB
external_benchmark_multitensor_allreduce.py	2.4 KB
external_benchmark_resnet.py	5.7 KB
external_benchmark_schedule.py	0.8 KB
external_cl_half_max.py	0.4 KB
external_hip_compiler_bug.py	10.2 KB
external_jit_failure.py	0.4 KB
external_llama_eval.py	3.6 KB
external_model_benchmark.py	7.1 KB
external_multi_gpu.py	2.6 KB
external_osx_profiling.py	1.2 KB
external_test_amd.py	1.2 KB
external_test_datasets.py	2.4 KB
external_test_embedding.py	0.2 KB
external_test_example.py	2.0 KB
external_test_hcq.py	16.8 KB
external_test_hip_compile.py	1.7 KB
external_test_hsa_driver.py	4.0 KB
external_test_image.py	1.5 KB
external_test_jit_on_models.py	1.5 KB
external_test_losses.py	0.8 KB
external_test_mamba.py	0.9 KB
external_test_metrics.py	0.8 KB
external_test_mnist_data_select.py	0.3 KB
external_test_nv.py	5.7 KB
external_test_onnx_backend.py	8.1 KB
external_test_opt.py	9.7 KB
external_test_optim.py	7.3 KB
external_test_speed_llama.py	2.3 KB
external_test_uops_graphing.py	1.3 KB
external_test_whisper_librispeech.py	2.7 KB
external_test_yolo.py	1.0 KB
external_test_yolov8.py	3.7 KB
fuzz_kfd.py	0.9 KB
fuzz_linearizer.py	9.9 KB
fuzz_schedule.py	4.4 KB
fuzz_shapetracker.py	2.4 KB
fuzz_shapetracker_math.py	2.2 KB
fuzz_symbolic.py	2.2 KB
fuzz_uops.py	3.5 KB
graph_batchnorm.py	1.8 KB
replay_schedule.py	1.2 KB
speed_beam_v_hcopt.py	1.7 KB
speed_compare_cuda_nv.py	2.7 KB
speed_compare_cuda_ptx.py	2.4 KB
verify_kernel.py	3.5 KB
/test/external/mlperf_bert/
external_test_checkpoint_loading.py	2.8 KB
/test/external/mlperf_bert/preprocessing/
create_pretraining_data.py	15.0 KB
external_test_preprocessing_part.py	5.1 KB
pick_eval_samples.py	4.0 KB
tokenization.py	12.7 KB
/test/external/mlperf_resnet/
lars_optimizer.py	9.3 KB
lars_util.py	6.6 KB
/test/external/mlperf_unet3d/
dice.py	3.4 KB
kits19.py	6.1 KB
/test/imported/
test_indexing.py	61.8 KB
/test/models/efficientnet/
car.jpg	8.1 KB
Chicken.jpg	110.2 KB
imagenet1000_clsidx_to_labels.txt	30.6 KB
/test/models/
test_bert.py	2.1 KB
test_efficientnet.py	3.1 KB
test_end2end.py	6.1 KB
test_mnist.py	3.9 KB
test_onnx.py	5.1 KB
test_real_world.py	6.2 KB
test_rnnt.py	1.7 KB
test_train.py	2.5 KB
test_waifu2x.py	0.7 KB
test_whisper.py	3.8 KB
/test/models/waifu2x/
input.png	7.3 KB
output.png	14.9 KB
/test/models/whisper/
test.wav	35.7 KB
test2.wav	85.1 KB
/test/testextra/
test_export_model.py	2.3 KB
test_lr_scheduler.py	6.0 KB
/test/unit/
test_device.py	1.7 KB
test_disk_cache.py	3.6 KB
test_disk_tensor.py	13.5 KB
test_flopcounter.py	2.5 KB
test_helpers.py	9.1 KB
test_shapetracker.py	25.7 KB
test_shapetracker_math.py	6.9 KB
test_shm_tensor.py	1.0 KB
test_symbolic.py	19.7 KB
test_view.py	1.8 KB
/test/web/
test_webgpu.js	1.8 KB
/tinygrad/
__init__.py	0.4 KB
device.py	10.4 KB
dtype.py	6.2 KB
function.py	9.6 KB
helpers.py	12.8 KB
lazy.py	13.4 KB
multi.py	11.4 KB
ops.py	7.1 KB
tensor.py	129.2 KB
/tinygrad/codegen/
kernel.py	38.2 KB
linearizer.py	27.6 KB
uops.py	21.9 KB
/tinygrad/engine/
graph.py	5.2 KB
jit.py	11.0 KB
realize.py	11.1 KB
schedule.py	18.5 KB
search.py	11.3 KB
/tinygrad/nn/
__init__.py	12.9 KB
datasets.py	0.5 KB
optim.py	6.8 KB
state.py	10.1 KB
/tinygrad/renderer/
__init__.py	2.7 KB
assembly.py	17.9 KB
cstyle.py	24.0 KB
llvmir.py	10.3 KB
/tinygrad/runtime/autogen/
amd_gpu.py	65.0 KB
comgr.py	38.6 KB
cuda.py	250.4 KB
hip.py	245.5 KB
hsa.py	270.1 KB
kfd.py	29.9 KB
nv_gpu.py	1.7 MB
opencl.py	82.7 KB
/tinygrad/runtime/driver/
hip_comgr.py	3.4 KB
hsa.py	7.2 KB
/tinygrad/runtime/graph/
clang.py	2.0 KB
cuda.py	5.3 KB
hcq.py	8.1 KB
hsa.py	10.0 KB
metal.py	4.6 KB
/tinygrad/runtime/
ops_amd.py	31.8 KB
ops_clang.py	1.5 KB
ops_cuda.py	10.8 KB
ops_disk.py	2.7 KB
ops_gpu.py	7.5 KB
ops_hsa.py	16.2 KB
ops_llvm.py	2.2 KB
ops_metal.py	5.8 KB
ops_npy.py	0.4 KB
ops_nv.py	37.6 KB
ops_python.py	10.9 KB
/tinygrad/shape/
shapetracker.py	6.3 KB
symbolic.py	16.7 KB
view.py	17.1 KB
Total files 499