Skip to content

Conversation

@OlaRonning
Copy link
Member

The Haiku module requires a set of params (net$params) with layer shapes and rng_state to run. ECS shouldn't try to map these to constraint space.

@fehiepsi fehiepsi merged commit f997da2 into pyro-ppl:master Feb 28, 2024
OlaRonning added a commit to aleatory-science/numpyro that referenced this pull request May 6, 2024
@OlaRonning OlaRonning deleted the fix/ecs_haiku branch February 21, 2025 11:29
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