Rebecca Chen
|
9176a9f23c
|
Silence some pytype errors.
PiperOrigin-RevId: 527929776
|
2023-06-02 18:04:36 +01:00 |
|
Peter Hawkins
|
82a347438f
|
Fix pytype failures related to teaching pytype about NumPy scalar types.
PiperOrigin-RevId: 519205179
|
2023-06-02 18:03:40 +01:00 |
|
Peter Hawkins
|
c051e6a51d
|
Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.
PiperOrigin-RevId: 511294746
|
2023-06-02 18:02:16 +01:00 |
|
Jake VanderPlas
|
6fcb84268e
|
Use jax.tree_util.tree_map in place of deprecated tree_multimap.
The latter is a simple alias of the former, so this change is a no-op.
PiperOrigin-RevId: 461229165
|
2022-07-24 17:53:28 +01:00 |
|
Trevor Thackston
|
32c1427c74
|
Remove conflicting version requirements
|
2021-08-06 01:12:17 -05:00 |
|
David Ding
|
3eddada361
|
Fix video_autoencoding colab reshape issue.
PiperOrigin-RevId: 388707196
|
2021-08-04 17:14:46 +01:00 |
|
David Ding
|
d96ef0b6ee
|
Download byte_tokenizers.py in Perceiver language modeling colab.
PiperOrigin-RevId: 388701647
|
2021-08-04 17:12:34 +01:00 |
|
David Ding
|
8cc5c39662
|
Bug fix for stitching function in optical flow colab.
PiperOrigin-RevId: 388465614
|
2021-08-04 17:10:23 +01:00 |
|
Drew Jaegle
|
21084c8489
|
Fix Perceiver IO author list, add correct ImageNet paper section.
PiperOrigin-RevId: 388423720
|
2021-08-03 15:20:19 +01:00 |
|
David Ding
|
887a4fcda6
|
Add links to colabs.
PiperOrigin-RevId: 388422650
|
2021-08-03 15:19:12 +01:00 |
|
David Ding
|
df9b857b4e
|
Add cell to install dependencies to run notebooks on cloud colab.
PiperOrigin-RevId: 388249525
|
2021-08-03 15:16:59 +01:00 |
|
David Ding
|
414683cc12
|
Add arXiv link for Perceiver IO.
PiperOrigin-RevId: 388165993
|
2021-08-03 15:13:37 +01:00 |
|
David Ding
|
ab87c9cd72
|
Follow common naming convention for all notebooks.
PiperOrigin-RevId: 387819214
|
2021-07-30 17:47:05 +01:00 |
|
David Ding
|
316297879f
|
Minor wording changes
PiperOrigin-RevId: 387817017
|
2021-07-30 17:45:58 +01:00 |
|
David Ding
|
768d530bf9
|
Add ImageNet classification colab.
PiperOrigin-RevId: 387805919
|
2021-07-30 17:44:52 +01:00 |
|
David Ding
|
12971d9a42
|
Example training pipeline for perceiver.
PiperOrigin-RevId: 387673066
|
2021-07-30 17:43:44 +01:00 |
|
Sebastian Borgeaud
|
8c431d40ea
|
Add the initial language colab for inference only
PiperOrigin-RevId: 387625457
|
2021-07-30 17:42:37 +01:00 |
|
David Ding
|
310620d045
|
Add optical flow visualization notebook.
PiperOrigin-RevId: 387615044
|
2021-07-30 17:41:30 +01:00 |
|
David Ding
|
595d0727b2
|
Add jax implementation of extract_patches.
PiperOrigin-RevId: 387610043
|
2021-07-30 17:40:22 +01:00 |
|
Sebastian Borgeaud
|
198cf845de
|
Move make_cross_attention_mask into the primitives section.
PiperOrigin-RevId: 387576037
|
2021-07-30 17:39:14 +01:00 |
|
Sebastian Borgeaud
|
0e552473e5
|
Add BytesTokenizer and EmbeddingDecoder
Both are required for the language colab (coming next)
PiperOrigin-RevId: 387574746
|
2021-07-30 17:38:05 +01:00 |
|
Andrew Lampinen
|
837430d4d1
|
Open sourcing the Ballet environment from "Towards Mental Time Travel..." which offers a challenging environment for remembering sequential events.
PiperOrigin-RevId: 387333851
|
2021-07-30 17:34:58 +01:00 |
|