Cornelius Wolff

icon indicating an email [email protected]

icon company Centrum Wiskunde & Informatica icon location Amsterdam, Netherlands icon location PhD Student at the Table Representation Learning (TRL) Lab in Amsterdam

Results 1 issues of Cornelius Wolff

[BUG] value_and_grad is really slow

12 comment

I am implementing a version of PPO in MLX and wanted to benchmark it against my PyTorch implementation. Sadly, the performance (samples per second) was really quite bad, so I...

performance