Search results

Torch2Jax vs Torchax?

Status: Open.

#32 In rdyro/torch2jax;

· adam-hartshorne opened on Nov 26, 2025

jax 0.7.0 Breaks Torch2Jax

Status: Open.

#31 In rdyro/torch2jax;

· adam-hartshorne opened on Jul 22, 2025

trace_graph fundamentally broken

Status: Open.

#30 In rdyro/torch2jax;

· RobertAgee opened on Jun 28, 2025

Jax 0.6.0 now requires cudnn 9.8

Status: Open.

#29 In rdyro/torch2jax;

· adam-hartshorne opened on Apr 24, 2025

[Question] jax.vmap & nested jax.vmap behavior

Status: Open.

#28 In rdyro/torch2jax;

· ZaberKo opened on Mar 24, 2025

pmap performance

Status: Open.

#25 In rdyro/torch2jax;

· rdyro opened on Jan 17, 2025

Multi-GPU Question

Status: Open.

#18 In rdyro/torch2jax;

· adam-hartshorne opened on Oct 27, 2024

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

Status: Open.

#6 In rdyro/torch2jax;

· adam-hartshorne opened on Aug 23, 2023