Issues 共 82
Any plans to upgrade with flax.nnx?
#118 · YinpeiDai · 2025-08-15
Too small initializer variance
#115 · LeoXinhaoLee · 2024-08-19
Unusual Initializer Range default
#114 · evanatyourservice · 2024-08-13
[Bug] Error in Evaluation
#111 · LeoXinhaoLee · 2024-03-18
Llama 7b Pretraining Dtype
#110 · LeoXinhaoLee · 2024-03-15
ERROR: Accessing retired flag 'jax_enable_async_collective_offload'
#109 · LeoXinhaoLee · 2024-03-15
How does `accumulate_gradient_steps` work?
#108 · VictorSanh · 2024-03-08
Multi host TPU training
#107 · kathir-ks · 2024-02-19
Mistral
#104 · peregilk · 2023-11-23
Serving errors: deprecated dependencies and structure error
#103 · sjw8793 · 2023-11-07