Skip to content

Klax Initializers#71

Merged
jaosch merged 12 commits intodevelopfrom
feature/icnn-weight-init
Dec 1, 2025
Merged

Klax Initializers#71
jaosch merged 12 commits intodevelopfrom
feature/icnn-weight-init

Conversation

@Drenderer
Copy link
Owner

@Drenderer Drenderer commented Sep 12, 2025

Extended the initializer scheme from JAX and added Hoedt et al. initialization.

Todo:

  • Add unit tests for _initializer.py functions
  • Improve the comparison script.

@Drenderer Drenderer self-assigned this Sep 12, 2025
@jaosch jaosch marked this pull request as ready for review December 1, 2025 12:40
@jaosch
Copy link
Collaborator

jaosch commented Dec 1, 2025

Some of the tests through a warning due to the default in klax.Initializer being set to jnp.float_, i.e., jnp.float64. But I think the PR is still ready to be merged into develop.

@jaosch jaosch merged commit d0b1f55 into develop Dec 1, 2025
1 check passed
@Drenderer Drenderer deleted the feature/icnn-weight-init branch January 27, 2026 10:43
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.

2 participants