Add cast to tensor for DTensor inputs for groupedmlp - Githubissues

databricks / megablocks

Apache License 2.0

1.11k stars 154 forks source link

Add cast to tensor for DTensor inputs for groupedmlp #71

Closed eracah closed 6 months ago