Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

EMLX Roadmap #5

Closed
samrat opened this issue Nov 22, 2024 · 6 comments
Closed

EMLX Roadmap #5

samrat opened this issue Nov 22, 2024 · 6 comments

Comments

@samrat
Copy link
Contributor

samrat commented Nov 22, 2024

EMLX Backend Callbacks Roadmap

NOTE: If you're grabbing one or more callbacks to tackle, please add a comment below so we don't have 2 people working on the same callback at the same time.

Pending

  • from_pointer
  • to_pointer
  • window_scatter_max
  • window_scatter_min

LinAlg:

  • conv
  • lu
  • qr
  • svd
  • triangular_solve
  • cholesky
  • eigh
  • solve
  • determinant

Not supported (just need to implement a nice error message):

  • count_leading_zeros
  • population_count
  • reduce
  • window_reduce

Done

Misc

  • to_batched
  • pad
  • reverse
  • put_slice
  • gather
  • phase
  • indexed_add
  • indexed_put
  • top_k
  • cbrt

Setup & Manipulation

  • init
  • backend_deallocate
  • backend_copy
  • backend_transfer

Creation & Generators

  • constant
  • from_binary
  • eye
  • iota
  • to_binary
  • inspect

Shape Operations

  • as_type
  • bitcast
  • reshape
  • squeeze
  • broadcast
  • transpose

Element Access & Manipulation

  • slice
  • concatenate
  • stack
  • select
  • take
  • take_along_axis

Element-wise Binary Operations

  • add
  • subtract
  • multiply
  • pow
  • remainder
  • divide
  • atan2
  • min
  • max
  • quotient
  • bitwise_and
  • bitwise_or
  • bitwise_xor
  • left_shift
  • right_shift

Element-wise Comparison Operations

  • equal
  • not_equal
  • greater
  • less
  • greater_equal
  • less_equal
  • logical_and
  • logical_or
  • logical_xor

Element-wise Unary Operations

  • abs
  • bitwise_not
  • ceil
  • conjugate
  • floor
  • negate
  • round
  • sign
  • real
  • imag
  • is_nan
  • is_infinity
  • logical_not

Reduction Operations

  • all
  • any
  • sum
  • product
  • reduce_max
  • reduce_min
  • argmax
  • argmin
  • window_sum
  • window_product
  • window_max
  • window_min
  • cumulative_sum
  • cumulative_product
  • cumulative_min
  • cumulative_max

Linear Algebra Operations

  • dot

Sorting & Search Operations

  • sort
  • argsort

FFT Operations

  • fft
  • ifft
  • fft2
  • ifft2

Miscellaneous Operations

  • all_close
  • clip
@cocoa-xu
Copy link
Member

Going to grab some quick ones for now:

  • greater
  • less
  • logical_and
  • logical_or

Probably will grab a few more in the evening.

@cocoa-xu
Copy link
Member

cocoa-xu commented Nov 22, 2024

grabbing

  • bitwise_and
  • bitwise_or
  • bitwise_xor
  • logical_not
  • negative
  • round
  • sign
  • floor
  • ceil
  • atan2
  • pow

and

  • divide
  • left_shift
  • right_shift
  • real
  • imag
  • conjugate
  • remainder

also

  • backend_deallocate
  • backend_copy
  • backend_transfer

@cocoa-xu
Copy link
Member

cocoa-xu commented Nov 22, 2024

  • is_nan
  • is_infinity
  • quotient
  • all
  • any
  • product
  • sigmoid

@cocoa-xu
Copy link
Member

cocoa-xu commented Nov 22, 2024

taking some reduction functions

  • argmax
  • argmin
  • cumulative_sum
  • cumulative_product
  • cumulative_min
  • cumulative_max

@cocoa-xu
Copy link
Member

cocoa-xu commented Nov 22, 2024

taking

  • concatenate
  • stack
  • select
  • take_along_axis

@polvalente
Copy link
Collaborator

Closing in favor of #48 and #49

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants