joelberkeley / spidr

Accelerated machine learning with dependent types
Apache License 2.0
73 stars 4 forks source link

Implement uniform and normal sampling in Idris #346

Open joelberkeley opened 1 year ago

joelberkeley commented 1 year ago

The implementations for uniform and normal sampling in XLA use the same XLA API we use in Idris, so there's little performance gain from using those APIs. IIRC they also have subtle bugs in them. It would also be nice to keep the Expr as simple as possible so it's easier to implement other backends for spidr in the future if we want to.