3 ms·
Yep i started off with trying to get it to work with pytorch (https://github.com/bkkaggle/lm-training-research-project/blob/master/train_pt.py https://github.co
by bkkaggle 6y ago
Yep i started off with trying to get it to work with pytorch (https://github.com/bkkaggle/lm-training-research-project/blob/master/train_pt.py https://github.com/bkkaggle/lm-training-research-project/blo...) then with pt-lightning but the whole 1 user VM per TPU board limitation in pytorch-xla 7-8 months ago made me switch over to TF
- punnerud 6y agoJust as Google want you to do. Within 3-5 years you will probably see a high price increase and no where to go.
- bkkaggle 6y agoheh. I've been using jax for a couple of months and its been a pretty nice replacement of both pt and tf. it feels like what a ml framework would look like if it were built around easy scaling and dev friendliness.