joelberkeley / spidr

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

Tensor tuples #327

Closed joelberkeley closed 7 months ago

joelberkeley commented 2 years ago

XLA allows tuples of tensors. These may allow us to compute more than one tensor from a single graph, in a single compile + run.

joelberkeley commented 7 months ago

completed by #384