Bryan
Bryan
Yeah, absolutely
Alright, I plan to make a separate implementation so it works better with how you render the page. Any chance you could assign this to me in the mean time?
Is there some reason why this hasnt been merged to master? It appears the pull request is fully functional and ready, no?
can also confirm its happening to me as well
Im trying to install via poetry and I find that there is an issue where Jax-Metal will install a version of jax that cannot be overwritten by specifying an additonal...
> @BeeGass Are you sure your `python3` is an `arm64` binary? I've been bitten by this more times than I care for. Try > > ```py > import platform >...
> @BeeGass, have you tried setting env **ENABLE_PJRT_COMPATIBILITY=1** to run jax-metal with jaxlib>0.4.26? Yeah still same behavior. Am told that the jax version needs to be equal to or higher...
> ```toml > [project] > dependencies = [ > "common-dep1", > "common-dep2", > "jax>=0.4.18,=0.4.18,
> It seems that also adding a type handler for `nnx.training.optimizer.OptVariable` is sufficient. I'd rather not need to look into the details of these classes to write serializers/deserializers though. @SandSnip3r...
@cgarciae Honestly great explanation. Based on this conversation I was able to make this. Its not perfect but this works pretty well. ```python class CheckpointManager: def __init__(self, checkpoint_dir: str, keep_n:...