issues
search
poets-ai
/
elegy
A High Level API for Deep Learning in JAX
https://poets-ai.github.io/elegy/
MIT License
469
stars
32
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
[Feature Request] Distributed strategies: Tensor Parallel and Pipeline Parallel
#254
rxng8
opened
2 months ago
0
[Bug] timeout not passed to data_transfer_fn during dataloading
#253
ScottAlexanderCameron
opened
1 year ago
0
use poetry-core
#252
dotlambda
opened
1 year ago
0
Switch to poetry-core
#251
fabaff
closed
1 year ago
0
Documentation/API reference not accessible via project website[Bug]
#250
geomlyd
opened
1 year ago
0
[Bug] elegy does not work with latest haiku version
#249
murphyk
opened
2 years ago
5
CSVLogger iteration over a 0-d array
#248
ScottAlexanderCameron
opened
2 years ago
0
Metrics ignore "on" keyword arg
#247
ScottAlexanderCameron
opened
2 years ago
0
[Bug] Elegy crash on GPU
#246
jiyuuchc
opened
2 years ago
2
TypeError: '_DimPolynomial' object cannot be interpreted as an integer
#245
mwitiderrick
opened
2 years ago
0
Python 3.10 support
#244
gabrielcnr
opened
2 years ago
0
Fixed optax import in README instructions for Flax
#243
tanaymeh
closed
2 years ago
0
LSTM
#242
mwitiderrick
opened
2 years ago
0
ValueError: The `target` has to be an integer tensor.
#241
mwitiderrick
opened
2 years ago
0
TypeError: dot_general() got an unexpected keyword argument 'use_v2'
#240
mwitiderrick
opened
2 years ago
0
AttributeError: module 'tensorflow.compiler.tf2xla.python.xla' has no attribute 'optimization_barrier'
#239
mwitiderrick
opened
2 years ago
0
[Bug] Example is broken on GPU
#238
CForgie
opened
2 years ago
1
Fix optax import in readme
#237
niklasschmitz
opened
2 years ago
0
Vnext
#236
cgarciae
opened
2 years ago
0
Using Elegy with Equinox?
#235
nalzok
opened
2 years ago
1
[Bug] Both API example notebooks have stopped working on Colab
#234
nalzok
opened
2 years ago
5
[Bug] Importing elegy results in a deep-lying AttributeError
#233
ericmjl
closed
2 years ago
3
Low-level-prototype
#232
cgarciae
closed
2 years ago
0
Fix DataParallel from_local
#231
cgarciae
closed
2 years ago
1
RFC: Elegy/Treex Ecosystem Next Versions
#230
cgarciae
opened
2 years ago
0
Pure functions in Elegy ?
#229
pablo2909
closed
2 years ago
5
Lazy load wandb
#228
cgarciae
closed
2 years ago
1
Figure out how to test the Wandb callback on CI
#227
cgarciae
opened
2 years ago
0
Donate model's memory buffer to jit/pmap functions.
#226
lkhphuc
closed
2 years ago
3
[Feature Request] Gradient Accumulation
#225
lkhphuc
closed
2 years ago
1
[Feature Request] Automatic mixed precision training.
#224
lkhphuc
opened
2 years ago
2
[Bug] Model summary does not display submodules in execution order.
#223
lkhphuc
opened
2 years ago
2
Docs typos
#222
lkhphuc
closed
2 years ago
0
Support creating Dataloader directly from HuggingFace dataset
#221
lkhphuc
closed
2 years ago
2
Weights and Biases Callback for Elegy
#220
soumik12345
closed
2 years ago
1
Improve CI
#219
cgarciae
closed
2 years ago
0
Handle distributed metrics with psum
#218
cgarciae
closed
2 years ago
0
Cleanup Documentation
#217
cgarciae
opened
2 years ago
0
[Bug] Problem with computing metrics
#216
organic-chemistry
closed
2 years ago
6
Random state within a Loss
#215
cgarciae
opened
2 years ago
0
Fix import errors without dev dependencies
#214
cgarciae
closed
2 years ago
1
Fix import error when Haiku is not installed
#213
cgarciae
closed
2 years ago
1
Updates Treex to 0.6.5
#212
cgarciae
closed
2 years ago
0
Add more metrics
#211
anvelezec
opened
2 years ago
0
Add more Layers
#210
cgarciae
opened
2 years ago
0
Develop R2 metric
#209
anvelezec
closed
2 years ago
0
Develop root mean squared error metric
#208
anvelezec
closed
2 years ago
0
Develop mean squared metric
#207
anvelezec
closed
2 years ago
0
Develop cosine similarity metric
#206
anvelezec
closed
2 years ago
0
Develop IOU metric
#205
anvelezec
closed
2 years ago
0
Next