Ecosyste.ms: Issues

An open API service for providing issue and pull request metadata for open source projects.

GitHub / jax-ml/jax issues and pull requests

#24756 - pure_callback using a threadpool to accelerate vmap

Issue - State: open - Opened by Joshuaalbert 7 days ago - 2 comments
Labels: enhancement

#24753 - Facing Scaling issue on cpu (arm and x86).

Issue - State: open - Opened by choudhary-devang 7 days ago
Labels: bug

#24751 - Allow setting default_device with platform names.

Pull Request - State: open - Opened by Stella-S-Yan 7 days ago - 1 comment
Labels: pull ready

#24750 - disable remat HLO pass by default

Pull Request - State: open - Opened by keshavb96 7 days ago - 1 comment

#24749 - cudaErrorSymbolNotFound : named symbol not found

Issue - State: open - Opened by amanjitsk 7 days ago - 3 comments
Labels: bug

#24748 - jnp.reshape: raise TypeError when specifying newshape

Pull Request - State: open - Opened by jakevdp 7 days ago

#24741 - [Mosaic] Fix canonicalize_extract op name.

Pull Request - State: closed - Opened by copybara-service[bot] 8 days ago

#24740 - TracerBoolConversionError

Issue - State: closed - Opened by hangita101 8 days ago - 4 comments
Labels: question

#24738 - debug export

Pull Request - State: closed - Opened by copybara-service[bot] 8 days ago

#24737 - `scipy.sparse.csgraph.connected_components` implementation

Issue - State: open - Opened by vboussange 8 days ago - 2 comments
Labels: enhancement

#24735 - [Mosaic] Fix handling of i1 splat constants

Pull Request - State: closed - Opened by copybara-service[bot] 8 days ago

#24721 - Expose the HloOpcode::kPartitionId in Jax.

Issue - State: closed - Opened by ptoulme-aws 9 days ago - 7 comments
Labels: enhancement

#24706 - [XLA] Make our LLVM usage more googley

Pull Request - State: open - Opened by copybara-service[bot] 9 days ago

#24705 - RFC: specify jit static args via Static annotation

Pull Request - State: open - Opened by jakevdp 9 days ago - 3 comments
Labels: enhancement

#24700 - register_dataclass: allow marking static fields via dataclass.field

Pull Request - State: closed - Opened by jakevdp 9 days ago
Labels: pull ready

#24696 - Bump hypothesis from 6.102.4 to 6.116.0

Pull Request - State: closed - Opened by dependabot[bot] 10 days ago - 1 comment
Labels: dependencies, python

#24682 - Disable 64 bit codegen to HLO for Neuron backend

Issue - State: open - Opened by ptoulme-aws 10 days ago - 7 comments
Labels: enhancement

#24674 - [mosaic_gpu] Scalar arguments to kernels.

Pull Request - State: closed - Opened by copybara-service[bot] 12 days ago

#24664 - RFC: add pytree-compatible dataclass decorator

Pull Request - State: open - Opened by jakevdp 13 days ago - 3 comments

#24663 - Add a GPU implementation of `lax.linalg.eig`.

Pull Request - State: open - Opened by copybara-service[bot] 13 days ago - 1 comment

#24657 - Fix argmin docstring to not say "maximum"

Pull Request - State: closed - Opened by dymil 13 days ago
Labels: pull ready

#24655 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 13 days ago

#24645 - [Mosaic GPU] Improve correctness of benchmarking scripts

Pull Request - State: closed - Opened by andportnoy 14 days ago
Labels: pull ready

#24602 - GPU Pallas `decode_attention` improvements

Pull Request - State: closed - Opened by rdyro 15 days ago
Labels: pull ready

#24561 - Plumb dot dimension numbers into TPU matmul op.

Pull Request - State: closed - Opened by copybara-service[bot] 17 days ago

#24560 - Move utility functions in build.py to utils.py

Pull Request - State: open - Opened by copybara-service[bot] 17 days ago

#24557 - Add base case to _device_id_to_logical.

Pull Request - State: closed - Opened by copybara-service[bot] 17 days ago

#24556 - Fix breaking PGLE test-cases

Pull Request - State: closed - Opened by copybara-service[bot] 17 days ago

#24554 - Remove int4 from jtu.dtypes.all_dtypes.

Pull Request - State: open - Opened by jakevdp 17 days ago

#24552 - [Pallas:MGPU] Fix flaky debug_print tests

Pull Request - State: closed - Opened by copybara-service[bot] 17 days ago

#24549 - MaxText Getting Started fails

Issue - State: open - Opened by learning-to-play 17 days ago - 1 comment
Labels: bug

#24548 - segmentation fault from a simple test on GCP vm

Issue - State: open - Opened by jiyuuchc 17 days ago - 2 comments
Labels: bug

#24547 - New function to allow easy paralelization over vmap()

Issue - State: open - Opened by MRiabov 18 days ago - 4 comments
Labels: enhancement

#24545 - fix jax.custom_gradient to allow closing over non-autodiff tracers

Pull Request - State: open - Opened by mattjj 18 days ago
Labels: pull ready

#24544 - ⚠️ Nightly upstream-dev CI failed ⚠️

Issue - State: open - Opened by github-actions[bot] 19 days ago - 2 comments
Labels: CI

#24543 - Update pre-commit workflow to cache on jax version

Pull Request - State: open - Opened by dfm 19 days ago
Labels: pull ready

#24542 - Fix lint at head

Pull Request - State: closed - Opened by dfm 19 days ago
Labels: pull ready

#24541 - return in finally swallows exceptions

Issue - State: open - Opened by iritkatriel 19 days ago - 1 comment
Labels: bug

#24538 - Add new jnp.cumulative_prod function.

Pull Request - State: closed - Opened by jakevdp 19 days ago
Labels: pull ready

#24537 - Add some missing jax.numpy documentation

Pull Request - State: closed - Opened by jakevdp 19 days ago
Labels: documentation, pull ready

#24536 - Limit self-hosted jobs to JAX main repo

Pull Request - State: open - Opened by MichaelHudgins 20 days ago

#24535 - #jax Adds a missing comma to Pallas Quickstart

Pull Request - State: closed - Opened by copybara-service[bot] 20 days ago

#24525 - Remove duplicate entry in CHANGELOG.

Pull Request - State: closed - Opened by gnecula 20 days ago
Labels: pull ready

#24524 - Re-enable jax2tf test for dot algorithm with stricter TF version check

Pull Request - State: closed - Opened by dfm 20 days ago
Labels: pull ready

#24523 - [ROCm] ci build fixes

Pull Request - State: closed - Opened by Ruturaj4 20 days ago - 1 comment
Labels: pull ready

#24519 - Add GC support for PyTreeRegistry.

Pull Request - State: open - Opened by copybara-service[bot] 20 days ago

#24516 - improve checkpoint / remat concreteness error with static_argnums

Pull Request - State: open - Opened by mattjj 20 days ago
Labels: pull ready

#24515 - Add composite test to jax.

Pull Request - State: open - Opened by copybara-service[bot] 20 days ago

#24514 - Clear extra_jit_context when exiting.

Pull Request - State: closed - Opened by copybara-service[bot] 20 days ago

#24513 - [Mosaic TPU] Remove extra check

Pull Request - State: closed - Opened by copybara-service[bot] 21 days ago

#24511 - improve concreteness error message in remat

Pull Request - State: closed - Opened by mattjj 21 days ago
Labels: pull ready

#24510 - Allow dot algorithms in default_matmul_precision config

Pull Request - State: open - Opened by dfm 21 days ago

#24507 - Finish 0.4.35 release by removing dead code

Pull Request - State: closed - Opened by copybara-service[bot] 21 days ago

#24505 - [jax2tf] Fixes a bad interaction between jax2tf.convert, TF, and call_tf

Pull Request - State: closed - Opened by gnecula 21 days ago - 1 comment
Labels: pull ready

#24500 - [shape_poly] Expands support for random.choice

Pull Request - State: closed - Opened by gnecula 21 days ago
Labels: pull ready

#24499 - reduce_sum support for int4, int2 and uint4, uint2

Pull Request - State: open - Opened by EdouardYvinec 21 days ago - 2 comments

#24495 - fix the wrong output of pallas attention kernel when q_len!=kv_len

Pull Request - State: closed - Opened by zinccat 21 days ago
Labels: pull ready

#24491 - Pallas Attention Kernel Wrong results when length of query != length of KV

Issue - State: closed - Opened by zinccat 21 days ago - 2 comments
Labels: bug