• 3
• 0
• 0

emergent_in_context_learning

• 3
• 0
• 0

dm_nevis

NEVIS'22: Benchmarking the next generation of never-ending learners

• 3
• 0
• 0

conformal_training

This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classifiers".

• 3
• 0
• 0

s6

• 3
• 0
• 0

abcei_mab

• 3
• 0
• 0

enn_acme

• 3
• 0
• 0

kfac-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

• 3
• 0
• 0

mctx

Monte Carlo tree search in JAX

• 3
• 0
• 0

offpolicy_selection_eslb

• 3
• 0
• 0

diplomacy

• 3
• 0
• 0

dd_two_player_games

• 3
• 0
• 0

xmanager

A platform for managing machine learning experiments

• 3
• 0
• 0

annealed_flow_transport

• 3
• 0
• 0

dm_c19_modelling

• 3
• 0
• 0

meltingpot

A suite of test scenarios for multi-agent reinforcement learning.

• 3
• 0
• 0

dmvr

• 2
• 0
• 0

launchpad

• 2
• 0
• 0

distrax

• 3
• 0
• 0

csuite

• 3
• 0
• 0

functa