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

#24871 - Add numpy.put_along_axis.

Pull Request - State: open - Opened by carlosgmartin about 10 hours ago - 2 comments

#24870 - Dedent your yields!

Pull Request - State: closed - Opened by copybara-service[bot] about 11 hours ago

#24869 - AOT errors when setting compile options

Issue - State: open - Opened by man2machine about 11 hours ago
Labels: bug

#24867 - Unsupported type in metal PJRT plugin with rng_bit_generator

Issue - State: open - Opened by dmarro89 about 11 hours ago
Labels: bug

#24866 - CUDA_ERROR_SYSTEM_NOT_READY: system not yet initialized

Issue - State: open - Opened by carlosgmartin about 12 hours ago
Labels: bug

#24865 - finalize deprecation of jax.clear_backends

Pull Request - State: open - Opened by jakevdp about 13 hours ago
Labels: pull ready

#24864 - jnp.logaddexp2: simplify implementation

Pull Request - State: open - Opened by jakevdp about 13 hours ago
Labels: pull ready

#24862 - Move the Control Flow section of Sharp Bits into its own tutorial.

Pull Request - State: open - Opened by emilyfertig about 14 hours ago
Labels: documentation

#24861 - [JAX/version] Support adding custom version suffix.

Pull Request - State: open - Opened by yliu120 about 14 hours ago

#24860 - Add GPU overflow bugfix (#24846) to changelog.

Pull Request - State: closed - Opened by dfm about 16 hours ago
Labels: pull ready

#24858 - cond error by using Tracer as a condition

Issue - State: open - Opened by 2300504237 about 18 hours ago - 4 comments
Labels: bug, needs info

#24857 - pjit handling of static_argnames is broken when using "dynamic" python strings.

Issue - State: open - Opened by etarassov about 19 hours ago - 5 comments
Labels: bug, needs info

#24856 - Remove GPU test with unreasonably large memory footprint.

Pull Request - State: closed - Opened by copybara-service[bot] about 19 hours ago

#24854 - Make sure compilation_cache.is_cache_used always returns a bool

Pull Request - State: closed - Opened by hurryabit about 22 hours ago
Labels: pull ready

#24852 - [Mosaic] Fix mask creation for packed sublanes

Pull Request - State: closed - Opened by copybara-service[bot] 1 day ago

#24851 - Fix pgle profiling, broken in previous change.

Pull Request - State: closed - Opened by copybara-service[bot] 1 day ago

#24848 - 7x7 `nnx.Conv` using `float32` parameter dtype overflows(?) to `nan` when sharded

Issue - State: closed - Opened by joshhansen 1 day ago - 1 comment
Labels: bug

#24847 - Add wraparound for 2x2x2 v5p

Pull Request - State: closed - Opened by copybara-service[bot] 1 day ago

#24843 - Wrong determinant results for large batch

Issue - State: closed - Opened by ChenAo-Phys 1 day ago - 5 comments
Labels: bug

#24842 - jax.scipy.linalg.toeplitz: support implicit batching

Pull Request - State: open - Opened by jakevdp 1 day ago
Labels: pull ready

#24841 - Bump actions/cache from 4.1.1 to 4.1.2

Pull Request - State: closed - Opened by dependabot[bot] 1 day ago
Labels: pull ready, dependencies, github_actions

#24840 - Bump hypothesis from 6.102.4 to 6.118.7

Pull Request - State: open - Opened by dependabot[bot] 1 day ago
Labels: dependencies, python

#24839 - [Mosaic GPU] Only run tests requiring sm90a on Hopper

Pull Request - State: closed - Opened by andportnoy 1 day ago
Labels: pull ready

#24838 - Deepcopy of pjit functions failing (or not supported)

Issue - State: open - Opened by jlperla 1 day ago
Labels: bug

#24831 - Improvements in the handling of tracing debug info

Pull Request - State: open - Opened by gnecula 2 days ago
Labels: pull ready

#24830 - support q offset w.r.t k/v in flash_attention function

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

#24829 - Bug in the latest libtpu nightly release

Issue - State: open - Opened by knyazer 2 days ago
Labels: bug

#24828 - Fix buggy and confusing logic in the C++/pjit caching path.

Pull Request - State: closed - Opened by dougalm 2 days ago
Labels: pull ready

#24826 - potential jax 0.4.35 release issue?

Issue - State: open - Opened by haohuanw 3 days ago - 1 comment
Labels: bug

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

Issue - State: open - Opened by github-actions[bot] 3 days ago - 1 comment
Labels: CI

#24823 - [shape_poly] Fix the handling of jvp(lax.sort)

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

#24822 - Cleanup more remnants of the jax.experimental.host_callback

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

#24821 - TPU with sharding: grpc initialization failure

Issue - State: closed - Opened by knyazer 3 days ago - 2 comments
Labels: bug

#24820 - Pattern match dot algorithm spec to preset name

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

#24818 - [shape_poly] Remove caching for the symbolic shape evaluator

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

#24814 - jnp.bincount: support boolean inputs

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

#24813 - bincount rejects bool

Issue - State: closed - Opened by carlosgmartin 4 days ago
Labels: bug

#24810 - Specialize ufunc.reduce for monoidal binary ufuncs.

Pull Request - State: open - Opened by jakevdp 4 days ago
Labels: pull ready

#24806 - Internal: create decorators for defining ufuncs

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

#24799 - pre-commit run --all failed for ruff and mypy

Issue - State: closed - Opened by apivovarov 5 days ago - 1 comment
Labels: bug

#24798 - Flaky test tests/fft_test.py::FftTest::testFftfreq5

Issue - State: open - Opened by apivovarov 5 days ago - 2 comments
Labels: bug

#24796 - Check mem_stats key in test_lax_full_like_efficient

Pull Request - State: open - Opened by apivovarov 5 days ago

#24794 - `jax.lax.DotAlgorithm` generates an unnecessary conversion on H100

Issue - State: closed - Opened by andportnoy 5 days ago - 3 comments
Labels: bug

#24793 - Fix parallel pgle-tests execution.

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

#24792 - Update some outdated syntax in FFI tutorial

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

#24790 - Add new CI scripts for running Pytests

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

#24788 - Add commands to run Bazel GPU (non-RBE) jobs

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

#24787 - Nightly builds on Mac ARM fail complex function numerical tests

Issue - State: open - Opened by hawkinsp 5 days ago - 1 comment
Labels: bug

#24786 - Don't perform size 0 slices into scipy rotations.

Pull Request - State: closed - Opened by hawkinsp 5 days ago - 1 comment
Labels: pull ready

#24784 - Disable the paged_attention test on TPU v5p.

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

#24783 - Disable for_loop_test on TPU v5p.

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

#24778 - (follow-up #23852) add missing `typename` keyword to work with `gcc`

Pull Request - State: closed - Opened by cainmagi 5 days ago - 1 comment
Labels: pull ready

#24777 - [Pallas] Increase test coverage of pl.dot.

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

#24776 - Automated Code Change

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

#24774 - don't warn on unused `type: ignore`

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

#24773 - Fix pre-commit to run on all files in CI

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

#24772 - Allow 64-bit output types from ffi_call regardless of enable_x64 flag

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

#24771 - Adding support for copy_p primitive to jet.

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

#24770 - jax.device_get: handle generic extended dtypes

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

#24769 - Skip flaky test on tpuv4

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

#24766 - Won't submit. Experimenting.

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

#24764 - Remove sharding from custom_root_test.

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

#24763 - Added ragged attention for MQA/MHA/GQA

Pull Request - State: closed - Opened by Rifur13 6 days ago

#24762 - `shard_map` doesn't work with `jnp.insert`

Issue - State: open - Opened by mrlazy1708 6 days ago
Labels: bug

#24759 - start an 'about the project' docs page

Pull Request - State: closed - Opened by froystig 6 days ago
Labels: pull ready

#24758 - Fix copy-paste typo

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

#24757 - [Pallas] Fix lowering tests for reduction ops

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