Search results

determine tensor shapes programmatically

feature 🚀New feature or requestNew feature or request

Status: Open.

#23 In ml-gde/jflux;

· SauravMaheshkar opened on Oct 15, 2024

Serialise nnx weights during cli run

feature 🚀New feature or requestNew feature or request

Status: Open.

#22 In ml-gde/jflux;

· SauravMaheshkar opened on Oct 15, 2024

port t5 and clip to nnx.Module

feature 🚀New feature or requestNew feature or request

Status: Open.

#21 In ml-gde/jflux;

· SauravMaheshkar opened on Oct 15, 2024

add gradient checkpointing

feature 🚀New feature or requestNew feature or request

Status: Open.

#20 In ml-gde/jflux;

· SauravMaheshkar opened on Oct 15, 2024