rdyro/torch2jax issues and pull requests

#26 - Multi gpu support

Pull Request - State: closed - Opened by rdyro 23 days ago

#25 - pmap performance

Issue - State: open - Opened by rdyro 26 days ago - 6 comments

#24 - wrap_torch2jax/cpp/main.h(86): error: namespace "xla::ffi" has no member "RemainingRets"

Issue - State: closed - Opened by Drama1234 about 1 month ago - 2 comments

#23 - small file

Pull Request - State: closed - Opened by rdyro 2 months ago

#22 - adding matrix to the ci

Pull Request - State: closed - Opened by rdyro 2 months ago

#21 - remove dead code (previous non new ffi lowering)

Pull Request - State: closed - Opened by rdyro 3 months ago

#20 - New FFI Interface Error Branch - "TypeError: 'list' object is not callable"

Issue - State: closed - Opened by adam-hartshorne 3 months ago - 3 comments

#19 - Donate Args Causing Repeated Spawning of Warning

Issue - State: closed - Opened by adam-hartshorne 4 months ago - 5 comments

#18 - Multi-GPU Question

Issue - State: open - Opened by adam-hartshorne 4 months ago - 7 comments

#17 - JAX v0.4.35 - xla_client.register_custom_call_target has been deprecated

Issue - State: closed - Opened by adam-hartshorne 4 months ago - 1 comment

#16 - New JAX Functionality For Calling "Foreign" Code

Issue - State: closed - Opened by adam-hartshorne 5 months ago - 5 comments

#15 - CUDA bug and JAX breaking change

Issue - State: closed - Opened by zwei-beiner 6 months ago - 4 comments

#14 - gcc version?

Issue - State: closed - Opened by rkruegs123 9 months ago - 2 comments

#13 - Minor API change

Issue - State: closed - Opened by zwei-beiner 10 months ago - 4 comments

#12 - Multi-gpu support

Issue - State: closed - Opened by zwei-beiner 10 months ago - 25 comments

#11 - DLPack API Changes

Issue - State: closed - Opened by adam-hartshorne 10 months ago - 2 comments

#10 - Use of new JAX Performance Flags?

Issue - State: closed - Opened by adam-hartshorne over 1 year ago - 2 comments

#9 - Handling situation of having two identically named packages

Issue - State: closed - Opened by adam-hartshorne over 1 year ago - 3 comments

#8 - Any thoughts on this torch2jax alternative?

Issue - State: closed - Opened by adam-hartshorne over 1 year ago - 1 comment

#7 - custom_call() args need to be updated to remove out_types

Issue - State: closed - Opened by danielpmorton over 1 year ago - 3 comments

#6 - JAX / XLA adding the importing external dlpack-aware Python arrays.

Issue - State: open - Opened by adam-hartshorne over 1 year ago - 1 comment

#5 - Ignore

Issue - State: closed - Opened by adam-hartshorne over 1 year ago

#4 - New bug when optimising function

Issue - State: closed - Opened by adam-hartshorne over 1 year ago - 4 comments

#3 - Issue when attempting to optimise using torch2jax function

Issue - State: closed - Opened by adam-hartshorne over 1 year ago - 8 comments

#2 - Quick question about Overview Code Examples