Skip to content

Commit 5d2e668

Browse files
Fix challenge 74: add missing Examples section and fix JAX starter comment
- Add required <h2>Example</h2> section to challenge.html (was missing, checklist requires Implementation Requirements, Example(s), Constraints) - Fix starter.jax.py comment: "on the GPU" → "on GPU" to match CLAUDE.md JAX template format Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent ec78e1f commit 5d2e668

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

challenges/medium/74_gpt2_block/challenge.html

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,13 @@
119119
<li><strong>Residual 2:</strong> \(\text{output} = x' + F\)</li>
120120
</ol>
121121

122+
<h2>Example</h2>
123+
<p>With <code>seq_len</code> = 4, <code>x</code> has shape (4, 768) and <code>output</code> has shape (4, 768).
124+
The <code>weights</code> buffer packs all 7,087,872 block parameters in the order described in the
125+
Weight Layout section below. Applying the transformer block transforms each token embedding through
126+
layer norm, multi-head self-attention with residual, layer norm, feed-forward network with residual,
127+
producing a new (4, 768) embedding tensor.</p>
128+
122129
<h2>Implementation Requirements</h2>
123130
<ul>
124131
<li>Use only native features (external libraries are not permitted)</li>

challenges/medium/74_gpt2_block/starter/starter.jax.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
import jax.numpy as jnp
33

44

5-
# x, weights are tensors on the GPU
5+
# x, weights are tensors on GPU
66
@jax.jit
77
def solve(x: jax.Array, weights: jax.Array, seq_len: int) -> jax.Array:
88
# return output tensor directly

0 commit comments

Comments
 (0)