Branden Vandermoon issues

Repositories
Issues
Comments

Results 1 issues of


                                            Branden Vandermoon

Convert DenseGeneral to NNX

# Description Finishing out [PR 1604](https://github.com/AI-Hypercomputer/maxtext/pull/1604) for @cgarciae Note: I updated the logits in `golden_data_grpo_default.jsonl` to the NNX values. The NNX logits don't match the Linen ones since the RNG...