learning_machine

This is a list of used for self learning. Each subdirectory is supposed to be self-contained project.

Name		Name	Last commit message	Last commit date
Latest commit History 53 Commits
.github/workflows		.github/workflows
aoc		aoc
dlpack		dlpack
docs		docs
export_test		export_test
flash_attn_speed		flash_attn_speed
hanq/hanq		hanq/hanq
jax-huggingface		jax-huggingface
jax_perf		jax_perf
llama_ref		llama_ref
play_with_torch2jax		play_with_torch2jax
spmd_sharding		spmd_sharding
spmd_sharding_jax		spmd_sharding_jax
spmd_sharding_torchax		spmd_sharding_torchax
torch_pallas		torch_pallas
zig		zig
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
basic_ops.py		basic_ops.py
custom_kernel_spmd.py		custom_kernel_spmd.py
equinox_with_torch.py		equinox_with_torch.py
module_with_scalar_state.py		module_with_scalar_state.py
save_grad.py		save_grad.py
test_aot_function.py		test_aot_function.py

Navigation Menu