google/uncertainty-baselines issues and pull requests

#1334 - Avoid use of deprecated xla_bridge.get_backend().platform

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

#1333 - jax.numpy.clip: update use of deprecated arguments.

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

#1332 - Fix pytype error

Pull Request - State: open - Opened by copybara-service[bot] about 2 months ago

#1331 - Checkpoints for Deterministic Models on CIFAR-10 and CIFAR-100

Issue - State: open - Opened by NathanielCogneaux 4 months ago - 1 comment

#1330 - Remove dependence on old flax PRNG compat mode.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#1329 - [numpy] Fix test failures under NumPy 2.1.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#1328 - ResNet50 BatchEnsemble much slower than expected: `Conv2DBatchEnsemble` less optimized than `Conv2D`?

Issue - State: open - Opened by arthur-thuy 6 months ago - 1 comment

#1327 - Replace deprecated `jax.tree_` functions with `jax.tree.`

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#1326 - Replace deprecated `jax.tree_` functions with `jax.tree.`

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#1325 - Replace deprecated jax.random.shuffle with jax.random.permutation

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#1324 - No matching distribution for tfds-nightly when trying to install the package

Issue - State: open - Opened by jonathansim 10 months ago - 1 comment

#1323 - Download checkpoint models from GCP

Issue - State: closed - Opened by NickHauptvogel 11 months ago

#1322 - Internal clean up.

Pull Request - State: closed - Opened by copybara-service[bot] 12 months ago - 1 comment

#1321 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#1320 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] about 1 year ago - 1 comment

#1319 - [JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu").

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

#1318 - [LSC] change uses of jax.random.KeyArray and jax.random.PRNGKeyArray to jax.Array

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

#1317 - [LSC] change uses of jax.random.KeyArray and jax.random.PRNGKeyArray to jax.Array

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

#1316 - ModuleNotFoundError: No module named 'official.nlp.bert'

Issue - State: open - Opened by tiendatAI over 1 year ago - 1 comment

#1315 - [LSC] Ignore incorrect type annotations related to jax.numpy APIs

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

#1314 - Open source privileged information model code.

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

#1313 - Open source approximate full marginalization and distillation with priviledged information methods.

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

#1312 - Migrate GenericFunction usages to PolymorphicFunction

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

#1311 - Enable gpu/tpu training with XManager.

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago - 1 comment

#1310 - No public description

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

#1309 - [JAX] Replace uses of `jnp.array` in types with `jnp.ndarray`.

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

#1308 - [JAX] Replace jax.numpy.NINF with -jax.numpy.inf.

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

#1307 - No public description

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago - 1 comment

#1306 - Fix a typo in colab link

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

#1305 - Update default hyperparameters.

Pull Request - State: open - Opened by copybara-service[bot] over 1 year ago - 1 comment

#1304 - Uses default method for tf.image.resize (bilinear).

Pull Request - State: closed - Opened by copybara-service[bot] over 1 year ago - 1 comment

#1303 - Open source colab for "A Simple Zero-shot Prompt Weighting Technique to Improve Prompt Ensembling in Text-Image Models"

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

#1302 - Open source Pegasus experiment setup for the ICLR OOD paper.

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

#1301 - add python pseudo code for relative Mahalanobis distance used in the ICLR paper of OOD for conditional language models.

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

#1300 - Internal.

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

#1299 - Migrate the use of deprecated `inspect.getargspec` to `inspect.getfullargspec`.

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

#1298 - Silence some pytype errors.

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

#1297 - Paracrawl HET experiment

Pull Request - State: open - Opened by copybara-service[bot] over 1 year ago - 1 comment

#1296 - Paracrawl baseline task

Pull Request - State: open - Opened by copybara-service[bot] over 1 year ago - 1 comment

#1295 - Open source ImageNet PI ResNet models.

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

#1294 - Open source ImageNet PI dataset and TRAM and no PI training scripts.

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

#1293 - Migrate files from /third_party/py/uncertainty_baselines/experimental/shoshin to /third_party/py/skai. Small fixes to adhere to /third_party/py build requirements.

Pull Request - State: open - Opened by copybara-service[bot] over 1 year ago - 1 comment

#1292 - Deprecates unused Cardiotoxicity dataset and MLP model.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1291 - Enable distributed training across multiple GPUs or TPUs.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1290 - Optimizer for AUPRC.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1289 - Solve one hot encoding being hardcoded to two classes.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1288 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1287 - The return of the score_batch() should be restricted to a tuple of two items at maximum

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1286 - Use ResNet50v2 by default.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1285 - tensorflow_federated no attribute 'Popen' error

Issue - State: open - Opened by yizirui almost 2 years ago

#1284 - Add two tower model

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1283 - Internal

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1282 - Add flag to config to load small images.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1281 - Extend binary evaluation metrics via OneVsRest Strategy.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1280 - Add general SKAI dataset.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1279 - Update SKAI dataset code.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1278 - Saves best model during training. Updates default behavior to saving model (more easily reproducible results) instead of checkpoints.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1277 - Update types to prevent pytype errors with an upcoming change to JAX.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1276 - Internal

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1275 - Suppress some pytype errors related to improved JAX types under pytype.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1274 - Open source CIFAR-10H Tensorflow Dataset construction and preprocessing.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1273 - plex_vit_demo : Failed to get url-imagenetv2public

Issue - State: open - Opened by Eezzeldin almost 2 years ago

#1272 - Can ignore some directory in t5 utils `latest_checkpoint_paths`.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1271 - Add two_tower_model.

Pull Request - State: open - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1270 - Assignment map for T5 het in the diagonal case.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1269 - Replaces references to jax.numpy.DeviceArray with jax.Array.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1268 - Update T5 het gin config.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1267 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1266 - Add covariance initial scale as argument of `HeteroscedasticDecoder`.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1265 - The heteroscedastic layer can reuse the pretrained head of T5 models.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1264 - Suppress some pytype errors related to jnp.DeviceArray == jax.Array.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1263 - Add in T5 the learnable temperature parameter of the het layer.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1262 - Surface more arguments of the `HeteroscedasticDecoder`.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1261 - [JAX] Fix up uses of PyTree and PyTreeDef types.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1260 - input_utils to importing in Colab notebook

Issue - State: open - Opened by JonoSax almost 2 years ago

#1259 - Update default behavior.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago - 1 comment

#1258 - #ISP Fix bugs in toxic_comment dataset argument passing, evaluation and inference

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1257 - Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1256 - Rename `in_axis_resources` and `out_axis_resources` with `in_shardings` and `out_shardings`. This is just a simple name replacement. It does not change any of the current pjit semantics and doesn't break any code.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1255 - Silence some pytype errors.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1254 - Rename `in_axis_resources` and `out_axis_resources` with `in_shardings` and `out_shardings`. This is just a simple name replacement. It does not change any of the current pjit semantics and doesn't break any code.

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1253 - #toxic_comments support filter data by created date

Pull Request - State: closed - Opened by copybara-service[bot] almost 2 years ago

#1252 - Question about SNGP BERT flags

Issue - State: closed - Opened by xuefei-wang about 2 years ago - 2 comments

#1251 - Replace uses of deprecated JAX sharding APIs with their new names in jax.sharding.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#1250 - Create a T5x experiment setup for BIG-Bench Multiple Choice Question Answering.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#1249 - Internal migration of Xmanager API.

Pull Request - State: open - Opened by copybara-service[bot] about 2 years ago - 1 comment

#1248 - Fix Imagenet ReaL masking bug.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#1247 - Internal.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#1246 - Implements upsampling for the most underrepresented group.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago - 1 comment

#1245 - Calculates weighted average accuracy, weighting each subgroup's acc by its proportionality.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago - 1 comment

#1244 - Update pytype disable markers for Python 3.10.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#1243 - Minor modififcations to CIFAR PI scripts.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#1242 - Fix imports in CIFAR-PI baselines

Pull Request - State: open - Opened by gortizji about 2 years ago

#1241 - Upgrade python from 3.7 to 3.8 in the CI to address failures from features from newer versions.

Pull Request - State: open - Opened by copybara-service[bot] about 2 years ago

#1240 - Adds support for hyperparameter sweeping with Vizier. Also adds a flag to control whether or not to make adjustments to the TF Keras ResNet model to be more like the PyTorch ResNet model.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago - 1 comment

#1239 - Catch and warn for import errors from TFDSBuilderFromSQLClientData.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#1238 - Fix types of ImageNet ReaL preprocessing.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#1237 - Zero-shot Clip Robustness and generalization Evaluation

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago - 1 comment

#1236 - Open source CIFAR-10N and CIFAR-100N privileged information datasets and no PI and TRAM baselines.

Pull Request - State: closed - Opened by copybara-service[bot] about 2 years ago

#1235 - RETINA: Data Loading Improvements, TF Cast Fix

Pull Request - State: open - Opened by nband about 2 years ago - 1 comment

GitHub / google/uncertainty-baselines issues and pull requests