Skip to content

Remove flax dependency#2

Merged
TheodoreWolf merged 3 commits intomainfrom
theo/remove-flax-dependancy
Apr 11, 2026
Merged

Remove flax dependency#2
TheodoreWolf merged 3 commits intomainfrom
theo/remove-flax-dependancy

Conversation

@TheodoreWolf
Copy link
Copy Markdown
Owner

@TheodoreWolf TheodoreWolf commented Apr 11, 2026

  • Removes flax as a core dependency
  • Adds jaxtyping to properly type the pytrees

@codecov
Copy link
Copy Markdown

codecov Bot commented Apr 11, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.

📢 Thoughts on this report? Let us know!

@TheodoreWolf TheodoreWolf merged commit 5fbfe13 into main Apr 11, 2026
4 checks passed
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