Refactoring Maxdiffusion-JAX-Stable-Stack Build Process

AI-Hypercomputer / maxdiffusion

Apache License 2.0

149 stars 17 forks source link

Closed parambole closed 2 months ago

parambole commented 2 months ago

Amending the workflow to build maxdiffusion with Jax-ss, which will then be used for XLML tests.
Refactored to replace with JAX-Stable-Stack