Skip to content

Add RMSNorm and rotary position embeddings - #55

Open
cetagostini wants to merge 2 commits into
pymc-devs:mainfrom
cetagostini:agent/rmsnorm-rope
Open

Add RMSNorm and rotary position embeddings#55
cetagostini wants to merge 2 commits into
pymc-devs:mainfrom
cetagostini:agent/rmsnorm-rope

Address review: simplify, fix false JAX claim, correct dtype handling

8aeb005
Select commit
Loading
Failed to load commit list.
Sign in for the full log view

Annotations

2 warnings
unittest (ubuntu-latest, 3.12, tests/ )
succeeded Aug 8, 2026 in 9m 57s