feat(BSS2): Add and enable jaxsnn

This adds jaxsnn to the ESD.

(Just testing for now…)

added 2 commits

• 0c211944 - feat(BSS2): add jaxsnn (event-based ML)
• a34e535c - feat(BSS2): enable jaxsnn

Compare with previous version

added 3 commits

• e302f2f5 - feat(jax): add new versions
• 6f328776 - feat(BSS2): add jaxsnn (event-based ML)
• 2f0779cb - feat(BSS2): enable jaxsnn

Compare with previous version

mentioned in merge request !498 (merged)

added 2 commits

• 5cd6aa98 - feat(BSS2): add jaxsnn (event-based ML)
• 897fd74f - feat(BSS2): enable jaxsnn

Compare with previous version

maybe check the test results, as in !498 (merged)? They don't look great

Test Summary:
   Errors...: 19
   Failures.: 0
   Skipped..: 0
   Total....: 19 

ACK, it's missing a dependency… (and some transient ones)

what I meant is, maybe also add some parsing of the test output?

Can I put the other MR below ;)? I.e. will you rebase before merging?

I'm not sure what below means, but yes, I will rebase

I'm a bit hesitant to "layer" changes on top of each other in MRs… (in my usual gerrit workflow I use it all the time, because changes might depend on other changes) — if a merge happens → ugly history with duplicated commits (even if it's a noop on the repo state); if it's a rebase, it's still weird to do so w/ gitlab. It just feels "hacky" to use it that way.

I meant I'm just not sure why the changes in this MR depend on the changes in !498 (merged)? (or you meant the other way round?)

True, we replicate the logic (bad!) in each package file, but semantically it contains the same things… → will add to this MR.

(Done (i.e. replicated code once again… :P — are you aware of a nice case of "common functionality" between packages in some other package (maybe upstream)?))

I haven't seen anyone do this, no

(but you can import from other package definitions, from spack.pkg.<repo>.<package> import ..., (or maybe even not necessarily a package definition, just a file in packages/?))

Hmm, maybe a dummy package that isn't buildable? Maybe slightly more beautiful?

sure, why not

added 6 commits

• 897fd74f...93c18642 - 2 commits from branch master
• 5491e4af - feat(jax): add new versions
• efcaa5e0 - feat: add py-optax
• 10b055b7 - feat(BSS2): add jaxsnn (event-based ML)
• 7d291b22 - feat(BSS2): enable jaxsnn

Compare with previous version

2/20 now, progress :)

   Errors...: 18
   Failures.: 0
   Skipped..: 0
   Total....: 20 

It's all the same error, I guess → version issue, checking.

py-chex too old… provided custom package

added 5 commits

• 692726de - 1 commit from branch master
• f334615e - feat(jax): add new versions
• b5933f64 - feat: add py-optax
• 36f06149 - feat(BSS2): add jaxsnn (event-based ML)
• a5c6d26c - feat(BSS2): enable jaxsnn

Compare with previous version

added 2 commits

• 0b066df8 - feat(BSS2): add jaxsnn (event-based ML)
• 9e36e3be - feat(BSS2): enable jaxsnn

Compare with previous version

added 2 commits

• bf6b005c - feat(BSS2): add jaxsnn (event-based ML)
• 6c0c924b - feat(BSS2): enable jaxsnn

Compare with previous version

added 2 commits

• ffa232a1 - feat(BSS2): add jaxsnn (event-based ML)
• 85367493 - feat(BSS2): enable jaxsnn

Compare with previous version

added 2 commits

• 0c4dc8c2 - feat(BSS2): add jaxsnn (event-based ML)
• 15d17514 - feat(BSS2): enable jaxsnn

Compare with previous version

added 2 commits

• ce7079d5 - feat(BSS2): add jaxsnn (event-based ML)
• 783a108d - feat(BSS2): enable jaxsnn

Compare with previous version

added 6 commits

• 783a108d...407893ff - 2 commits from branch master
• be0a42ef - feat(jax): add new versions
• 7295d04d - feat: add py-optax
• 65922878 - feat(BSS2): add jaxsnn (event-based ML)
• 25eb3601 - feat(BSS2): enable jaxsnn

Compare with previous version

@emuller looks ok now, should I merge this?

marked this merge request as ready

ACK, just pylint/pycodestyle fails… that's what we expect :D

mentioned in merge request !512 (merged)

added 5 commits

• 3fbf807e - 1 commit from branch master
• f7c366b8 - feat(jax): add new versions
• 5e3eb805 - feat: add py-optax
• 923deb59 - feat(BSS2): add jaxsnn (event-based ML)
• 156f31d4 - feat(BSS2): enable jaxsnn

Compare with previous version

merged