issues
search
stanford-crfm
/
haliax
Named Tensors for Legible Deep Learning in JAX
Apache License 2.0
140
stars
9
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
`jax.lax.switch` for Haliax
#46
anh-tong
closed
8 months ago
2
Add `dslice` for dynamic slices, patterned off pallas
#45
dlwh
closed
8 months ago
0
Add einops-style `rearrange`
#44
dlwh
closed
8 months ago
0
How does haliax work with mixed precision?
#43
samuelstevens
closed
8 months ago
3
Add Conv and ConvTranspose
#42
dlwh
closed
9 months ago
2
Release 1.1
#41
dlwh
closed
9 months ago
0
add safe_wraps for wrapping fns or Modules
#40
dlwh
closed
9 months ago
0
Lowerable Named Jit
#39
dlwh
closed
9 months ago
0
added custom wrapper for jax.lax.top_k in new file
#38
rohan-mehta-1024
closed
10 months ago
0
Add context to help fix `Array`s that are not fully addressable` errors
#37
dlwh
opened
10 months ago
5
In-place Array Modificatons
#36
bmac3
closed
10 months ago
1
Move `state_dict` serialization to Haliax
#35
dlwh
opened
10 months ago
17
`haliax.debug.diagnose_common_issues`
#34
dlwh
closed
10 months ago
0
added implementation and tests for ravel function
#33
rohan-mehta-1024
closed
10 months ago
0
Worth adding top_k function?
#32
rohan-mehta-1024
closed
9 months ago
3
Update __bool__() in core.py
#31
Ivan-Zhou
closed
10 months ago
3
better support for the `inference` field in dropout
#30
dlwh
closed
10 months ago
0
Use `inference` field a la equinox
#29
dlwh
closed
10 months ago
0
Make a better error message for abstract tracer errors
#28
dlwh
opened
10 months ago
0
RNN: Add RNN Cells
#27
dlwh
opened
10 months ago
0
NN Layers: RNN Backbone
#26
dlwh
opened
10 months ago
0
NN Layers: Add Pooling
#25
dlwh
closed
6 months ago
1
NN Layers: Add GroupNorm
#24
dlwh
opened
10 months ago
0
NN Layers: Convolutions
#23
dlwh
closed
9 months ago
0
Add more NN Layers
#22
dlwh
opened
10 months ago
0
add hypothesis tests (mostly testing shapes)
#21
dlwh
opened
10 months ago
0
Maybe warn if `dot` reduces a dim that isn't in common between args?
#20
dlwh
opened
10 months ago
0
Add einops-style `rearrange`
#19
dlwh
closed
8 months ago
1
Add `ravel` / `flatten`
#18
dlwh
closed
10 months ago
3
Why init instead of __init__?
#17
rohan-mehta-1024
closed
10 months ago
2
Add lots of docs for Haliax, support for readthedocs
#16
dlwh
closed
10 months ago
0
support fancier masks in levanter fa branch
#15
dlwh
closed
11 months ago
0
Batched Take
#14
dlwh
closed
11 months ago
0
Why do all axes have to be unique?
#13
rohan-mehta-1024
closed
11 months ago
3
FSDP function for less boilerplate
#12
dlwh
closed
11 months ago
0
Various changes in support of flash attention
#11
dlwh
closed
11 months ago
0
Model Inference with Haliax?
#10
rohan-mehta-1024
closed
5 months ago
5
Update README.md with new tutorials
#9
dlwh
closed
12 months ago
0
Tutorial Suggestion
#8
pharringtonp19
closed
1 year ago
2
Upgraded to latest Equinox; removed Chex dependency.
#7
patrick-kidger
closed
1 year ago
1
Run pre commit in CI
#6
dlwh
closed
1 year ago
0
Support PrefixLM attention mask
#5
dlwh
closed
1 year ago
0
Pipeline Parallelism in `Stacked`
#1
dlwh
opened
1 year ago
1
haliax.dot: allow defining output order a la einsum
#2
dlwh
opened
1 year ago
8
Equinox woes: declared class attributes
#3
dlwh
opened
1 year ago
0
Make Haliax docs
#4
dlwh
closed
10 months ago
4
Previous