search

d2l-ai / d2l-en

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
https://D2L.ai
Other
23.83k stars 4.35k forks source link

SageMaker Studio Lab link is outdated and is only for PyTorch #2508

Closed AngelynDisguise closed 1 year ago

AngelynDisguise commented 1 year ago

I started trying out Amazon SageMaker Studio Lab recently, image

and the button on D2L links you to: https://studiolab.sagemaker.aws/import/github/d2l-ai/d2l-pytorch-sagemaker-studio-lab/blob/main/GettingStarted-D2L.ipynb

This makes you clone this project: d2l-pytorch-sagemaker-studio-lab (https://github.com/d2l-ai/d2l-pytorch-sagemaker-studio-lab) image

I noticed that this repo hasn't been updated since 2022 and even though I clicked on JAX as my preference on D2L, the SageMaker Studio Lab button still brings me to this PyTorch repo. Here's an image where you can clearly see the difference between D2L (on the left) and this repo (on the right). image

I am currently following along the d2l-jax-sagemaker repo (https://github.com/d2l-ai/d2l-jax-sagemaker), and it seems up to date. But I would really appreciate it if the button on the actually D2L site actually brought us to an updated D2L repo for Studio Lab, and corresponds to a ML framework preference. Studio Lab seems like a cool environment to learn ML and I would like to follow D2L in JAX.

AnirudhDagar commented 1 year ago

You can use the latest version of jax notebooks by cloning the jax sagemaker repo from https://github.com/d2l-ai/d2l-jax-sagemaker Note that you will have to setup the studio lab environment manually (installing the correct version of jax, flax etc. you can check https://github.com/d2l-ai/d2l-en/blob/master/static/build.yml for the supported versions) in this case since we don't automatically set that up for frameworks other than pytorch.