Skip to content

Linen set actions#33

Merged
smorad merged 4 commits into
mainfrom
linen-set-actions
Jun 1, 2026
Merged

Linen set actions#33
smorad merged 4 commits into
mainfrom
linen-set-actions

Conversation

@smorad

@smorad smorad commented Jun 1, 2026

Copy link
Copy Markdown
Owner

This implements all of the equinox set actions in linen. We also update the linen modules to use the equinox.nn.Linear init by default, which is identical to the linear init in torch. We find this performs slightly better than the default flax init, and our perf tests will not pass without it.

smorad and others added 4 commits June 1, 2026 11:48
Move runtime deps and extras into pyproject.toml and configure setuptools build backend.

Co-authored-by: Cursor <cursoragent@cursor.com>
Check in uv.lock + requirements snapshot and document uv-based installs alongside pip.

Co-authored-by: Cursor <cursoragent@cursor.com>
@smorad
smorad merged commit 2041bff into main Jun 1, 2026
1 check passed
@smorad
smorad deleted the linen-set-actions branch June 1, 2026 08:59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant