Closed mortont closed 8 months ago
Looks like this issue can be closed @mortont as #12 is merged?
I still need to add concatenate
before closing this, should be pretty straightforward. PRs welcome as well!
With #22 closed, I'm going to close this issue. If we want to support concatenating tensors of differing types I'd suggest a new issue. Right now, that would be a seldom (if ever) use case for Ortex since concatenation is typically only used during Nx.Serving.batched_run
on tensors of the same type.
Having these will reduce the number of
Nx.backend_transfer
calls needed when using Ortex with autoregressive models. There are likely others, similar to https://github.com/elixir-nx/ortex/issues/8 that will make this more ergonomic.