{"library":"tensorflow","install":[{"cmd":"pip install tensorflow","imports":["# Option 1: Use standalone Keras 3 (recommended for new code)\nimport keras\nmodel = keras.Sequential([keras.layers.Dense(64, activation='relu')])\n\n# Option 2: Access via tf.keras (same Keras 3 in TF 2.16+)\nimport tensorflow as tf\nmodel = tf.keras.Sequential([tf.keras.layers.Dense(64)])","@tf.function\ndef train_step(x, y):\n    with tf.GradientTape() as tape:\n        pred = model(x, training=True)\n        loss = loss_fn(y, pred)\n    grads = tape.gradient(loss, model.trainable_variables)\n    optimizer.apply_gradients(zip(grads, model.trainable_variables))\n    return loss"]},{"cmd":"pip install tensorflow[and-cuda]","imports":[]},{"cmd":"pip install tensorflow-cpu","imports":[]},{"cmd":"pip install tf-keras","imports":[]}]}