Taekyoon Ted Choi

Taekyoon

AI & ML interests

NLP

Recent Activity

updated a dataset 8 days ago
Taekyoon/test_amazon
published a dataset 8 days ago
Taekyoon/test_amazon
updated a model 6 months ago
Taekyoon/k_qwen_1_5
View all activity

Organizations

None yet

Taekyoon's activity

reacted to beomi's post with 👍 9 months ago
view post
Post
14376
#TPU #PyTorch #Jax

When You're trying to use PyTorch or Jax on TPU,

for v2/v3/v4:
use tpu-ubuntu2204-base

for v5p:
use v2-alpha-tpuv5

for v5e:
use v2-alpha-tpuv5-lite

You must use these base images for the system to 'boot'.

Previously used tpu-vm-v4-pt-1.13 images might seem to start the VM, but SSH connections do not work.

I thought it was a firewall issue and spent a lot of time on it before realizing it was a problem with the boot image 🥲

https://cloud.google.com/tpu/docs/runtimes#pytorch_and_jax
New activity in beomi/gemma-mling-7b 10 months ago

Update README.md

#4 opened 10 months ago by
Taekyoon

Update README.md

#3 opened 10 months ago by
Taekyoon

Update metric table

#2 opened 10 months ago by
Taekyoon