GitHub - sdrshn-nmbr/inkle: Thinking Machines Inkling inference on TPUs via JAX — checkpoint IO, NVFP4 decode, CPU oracle, v5e streaming runner · GitHub

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
.gitignore		.gitignore
checkpoint_inventory.py		checkpoint_inventory.py
checkpoint_io.py		checkpoint_io.py
cpu_reference.py		cpu_reference.py
inkling_layout.py		inkling_layout.py
inspect_released_model.py		inspect_released_model.py
jax_layer_probe.py		jax_layer_probe.py
nvfp4_expert_jax.py		nvfp4_expert_jax.py
real_expert_jax.py		real_expert_jax.py
run_gcp_tpu.sh		run_gcp_tpu.sh
streaming_tpu_inference.py		streaming_tpu_inference.py
test_checkpoint_io.py		test_checkpoint_io.py
tiny_jax.py		tiny_jax.py
tiny_quantization_tpu.py		tiny_quantization_tpu.py
tiny_reference.py		tiny_reference.py
tiny_tpu.py		tiny_tpu.py
verify_released_expert.py		verify_released_expert.py

About

Thinking Machines Inkling inference on TPUs via JAX — checkpoint IO, NVFP4 decode, CPU oracle, v5e streaming runner

0 stars

0 watching

Report repository

Contributors

Languages