-
Notifications
You must be signed in to change notification settings - Fork 2.8k
Issues: jax-ml/jax
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
Author
Label
Projects
Milestones
Assignee
Sort
Issues list
+ดูหนังออนไลน์! วัยสตาร์ท น็อนสต็อป (2024) เต็มเ รื่องฟรี ซับไทย-พากย์ ไทย สตรีมและดาวน์โหลดคุณภาพสูง!
enhancement
New feature or request
#25656
opened Dec 22, 2024 by
ghost
Add mode='fan_geom_avg' to nn.initializers.variance_scaling
enhancement
New feature or request
#25649
opened Dec 20, 2024 by
carlosgmartin
Improvements to debug_nans
enhancement
New feature or request
#25643
opened Dec 20, 2024 by
emilyfertig
5 tasks
Add New feature or request
mask
argument to lax.argmax
enhancement
#25623
opened Dec 20, 2024 by
carlosgmartin
Add New feature or request
replace: bool
argument to random.categorical
to sample without replacement using Gumbel-top-k trick
enhancement
#25617
opened Dec 19, 2024 by
carlosgmartin
Surprising difference of output between NumPy's Something isn't working
float32
and JAX's float32
bug
#25601
opened Dec 19, 2024 by
jeertmans
error: 'string' in namespace 'std' does not name a type
bug
Something isn't working
#25598
opened Dec 19, 2024 by
adamjstewart
Error when following the JAX tutorial on "Introduction to parallel programming"
bug
Something isn't working
#25541
opened Dec 17, 2024 by
Stella-S-Yan
Make New feature or request
xla_bridge.is_gpu
more extensible
enhancement
#25521
opened Dec 17, 2024 by
PragmaTwice
Clarification re: supported data types in New feature or request
jax.linearize
and jax.linear_transpose
enhancement
#25517
opened Dec 16, 2024 by
johannahaffner
Public API for mesh axis index size
enhancement
New feature or request
#25515
opened Dec 16, 2024 by
gabbard
jax.random.choice(replace=True) samples 0 probability index
bug
Something isn't working
duplicate
This issue or pull request already exists
#25498
opened Dec 16, 2024 by
LemonATsu
jax.scipy.spatial.transform.Rotation.from_quat is missing scalar_first flag
bug
Something isn't working
#25491
opened Dec 14, 2024 by
adam-hartshorne
register_dataclass
does not handle __init__
methods
bug
#25486
opened Dec 13, 2024 by
patrick-kidger
Slow convolution, many memory warnings
bug
Something isn't working
#25461
opened Dec 13, 2024 by
apasarkar
Segmentation fault in jaxlib 0.4.37 and Debian 11 ARM64
bug
Something isn't working
#25436
opened Dec 12, 2024 by
gongomgra
Replace np.array() with np.fromfile() to improve performance
enhancement
New feature or request
#25418
opened Dec 11, 2024 by
perfNerdJK
Latency Hiding Scheduler not working with jax 0.4.35
bug
Something isn't working
#25404
opened Dec 11, 2024 by
lgiacomoni
jax.full allocates memory on the wrong device
bug
Something isn't working
#25396
opened Dec 11, 2024 by
Chaosruler972
Unable to parse backend config for custom call: Could not convert JSON string to proto: Expected : between key:value pair.
bug
Something isn't working
#25389
opened Dec 10, 2024 by
adam-hartshorne
Pallas Kernel Expected Output Shape Error Using Grids On TPU
bug
Something isn't working
type:Bug
#25379
opened Dec 10, 2024 by
dshalem
Add syntax highlighting for Lowered and Compiled objects
enhancement
New feature or request
#25366
opened Dec 10, 2024 by
carlosgmartin
Error when donating buffer with under-specified sharding
bug
Something isn't working
#25346
opened Dec 9, 2024 by
jaro-sevcik
Lowering rule for New feature or request
lax.scan
unnecessarily emits while
loop for unroll=1
when length
or the size of xs
is 1
enhancement
#25330
opened Dec 7, 2024 by
carlosgmartin
[Pallas] Add lowering support for tl.join
enhancement
New feature or request
pallas
Issues pertaining to Pallas (GPU or TPU)
#25321
opened Dec 6, 2024 by
ggengnv
Previous Next
ProTip!
Find all open issues with in progress development work with linked:pr.