PufferLib
PufferLib copied to clipboard
Logits
Should hopefully be faster. Based on my comparison of different categorical distribution sampling methods. Wandb tests