mlx

mlx copied to clipboard

Support for einops

6

Hi :) I am a big fan of **einops** and I believe that supporting it within the framework would be highly valuable, especially for quick prototyping and situations where performance...

gboduljak

add support for tile

1

## Proposed changes `np.tile` function in mlx as a op `mlx.core.tile` ## Checklist Put an `x` in the boxes that apply. - [x] I have read the [CONTRIBUTING](https://github.com/ml-explore/mlx/blob/main/CONTRIBUTING.md) document -...

Bahaatbb

`repr` and `str` for bool arrays should print `True` and `False` instead of `true` and `false`

5

This is definitely low priority, but maybe worth fixing in the future. [`__repr__`](https://docs.python.org/3/reference/datamodel.html#object.__repr__): > If at all possible, this should look like a valid Python expression that could be used...

dastrobu

Why not implement this in Pytorch?

29

ToniRV

Additoinal losses

3

## Proposed changes Added commonly used losses with tests ## Checklist Put an `x` in the boxes that apply. - [x] I have read the [CONTRIBUTING](https://github.com/ml-explore/mlx/blob/main/CONTRIBUTING.md) document - [x] I...

Jyun1998

Official R plans?

2

In the [R](https://www.r-project.org) world, there is a very popular and efficient [Rcpp](https://www.rcpp.org) package, that provides a great level of integration of C++ projects into R universe. For example, the [torch](https://torch.mlverse.org)...

cyrusmsk

alibi error if log2(num_heads) is not an integer

4

hf alibi reference: https://github.com/huggingface/transformers/blob/a7cab3c283312b8d4de5df3bbe719971e24f4281/src/transformers/models/bloom/modeling_bloom.py#L82 axlearn alibi reference: https://github.com/apple/axlearn/blob/b92f666f661e6bacd757d7a37f1691d4f8985655/axlearn/common/adapter_torch.py#L561 https://github.com/apple/axlearn/blob/b92f666f661e6bacd757d7a37f1691d4f8985655/axlearn/common/attention.py#L3536 current mlx: https://github.com/ml-explore/mlx/blob/44c1ce5e6af2625571cd384e5be49e9778770ffc/python/mlx/nn/layers/positional_encoding.py#L184

gyin94

Unit test Floating Point Exception

12

Building from source and running on an M1 I am getting a floating point exception from the `Negatie slice and ascending bounds` portion of the `test_array.TestArray.test_slice_negative_step` unit test. All tests...

ccostes

PyTorch (MPS) is faster than MLX for training and inference for ResNets and Transformers (tested on 2 tasks)

16

Hi! (I was originally going to file this issue on `mlx-examples`, but I figured performance is more relevant to this repository.) I have had the chance to do some comparative...

SarthakYadav

## Proposed changes Add QR factorization primitive TODO: - [ ] impl for other dtypes - [ ] tests - [ ] python bindings potential tests: 1. reconstruction 2. orthogonality...

nullhook