lab
lab copied to clipboard
wesselb
A generic interface for linear algebra backends
Hi Is it possible to install wesselb/lab directly using conda? Do you provide packages for any conda repo? Thanks Chris.
Hi, We've been having (at geometric_kernels) some issues with people forgetting to import modules with definitions for backend-specific functions, as in `import geometric_kernels.torch`, for example. Currently the error messages are...
Hey Wessel! I am just wondering if it might be possible to add support for the MPS backend in PyTorch as an alternative to Cuda, which would allow for the...
This is a feature request to support [numpy's Generator](https://numpy.org/doc/stable/reference/random/generator.html#numpy.random.Generator) in addition to (legacy) `np.random.RandomState`.
`eye` doesn't keep tensors on the gpu: ```python In [1]: import lab as B In [2]: import lab.torch In [3]: import torch In [4]: B.eye(torch.ones((2, 2)).cuda()) Out[4]: tensor([[1., 0.], [0.,...
With the previous versions of plum and lab (`backends==1.4.32 plum-dispatch==1.7.4`), the following was a way in which one could register a new set of types for a function: ```python import...