Skip to content

feat: operator params introspection, eqx.Module tests, tutorial, and Linear.__eq__ fix - #161

Merged
astanziola merged 1 commit into
mainfrom
feat/operator-params-guide
Mar 22, 2026
Merged

astanziola merged 1 commit into
mainfrom
feat/operator-params-guide

Conversation

@astanziola

@astanziola astanziola commented Mar 22, 2026 •

Copy link
Copy Markdown
Member

Summary

  • Fix Linear.__eq__ crash when compared with non-Linear types (closes Type object 'type' has no attribute 'domain' #145)
  • Add get_implementations() and has_implementation() to jaxdf.util for operator dispatch introspection
  • Add tests verifying eqx.Module as operator params works with jit, grad, and vmap
  • Add Operator Parameters Guide tutorial notebook with visualizations on a real image, covering: param inspection, override, sharing, composition, neural networks as params, NN-generated stencils, and trained gradient correction
  • Update mkdocs nav and API docs for new functions
  • Increase pre-commit large file limit to 10MB for notebooks with embedded plots

All 331 tests pass locally. Notebook executes cleanly.

@astanziola
astanziola merged commit 5150d6d into main Mar 22, 2026
7 checks passed
@codecov

codecov Bot commented Mar 22, 2026

Copy link
Copy Markdown

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 93.43%. Comparing base (cd3bf1c) to head (ba5e7de).
⚠️ Report is 2 commits behind head on main.

Additional details and impacted files
@@            Coverage Diff             @@
##             main     #161      +/-   ##
==========================================
+ Coverage   93.34%   93.43%   +0.09%     
==========================================
  Files          15       15              
  Lines         931      945      +14     
==========================================
+ Hits          869      883      +14     
  Misses         62       62              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

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.

Type object 'type' has no attribute 'domain'

1 participant