Skip to content

refactor: Add shared specs and behaviors for Dense layers#13

Open
JulianSlzr wants to merge 5 commits into
mlx-abstract-simplefrom
mlx-abstract-dense
Open

refactor: Add shared specs and behaviors for Dense layers#13
JulianSlzr wants to merge 5 commits into
mlx-abstract-simplefrom
mlx-abstract-dense

Conversation

@JulianSlzr
Copy link
Copy Markdown
Collaborator

  • Add MLX dense.py implementations ported from DBraun's branch
  • Abstract Dense layers into specs/ and extract shared behavior tests into dense_behaviors.py
  • Standardize both JAX and MLX tests to inherit from and use shared specs for Dense layers

JulianSlzr and others added 5 commits April 18, 2026 22:54
Co-authored-by: David Braun <2096055+DBraun@users.noreply.github.com>
- Dense and EinsumDense use __init__(config) pattern with deferred init
- Removed _DenseEager wrapper, using dynamic submodule creation directly
- Added .copy() with @OverRide to backend Config classes
- Exported Dense/EinsumDense from mlx/__init__.py
- Updated behavior tests for deferred-init compatibility
@JulianSlzr JulianSlzr requested a review from rryan April 24, 2026 18:58
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