Skip to content

Add flax implementations for semigroups#36

Open
smorad wants to merge 1 commit into
mainfrom
flax-semi
Open

Add flax implementations for semigroups#36
smorad wants to merge 1 commit into
mainfrom
flax-semi

Conversation

@smorad

@smorad smorad commented Jun 8, 2026

Copy link
Copy Markdown
Owner

This ports the equinox semigroups over to flax and makes small updates to the tests to ensure both equinox/flax have the same test setup. We are experiencing some issues with the flax TTTL model, which might be related to a bad random seed. Nonetheless, we should keep it disabled for now.

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