Pinned Repositories
brax
Massively parallel rigidbody physics simulation on accelerator hardware.
flax
Flax is a neural network library for JAX that is designed for flexibility.
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
acme
A library of reinforcement learning components and agents
HierQ
Hierarchical multistep and Q(λ)
jit_env
An Agent-Environment interface with Jax transformation compatibility
PyCartogram
Produce Cartograms in Python with Shapely, Geopandas and numpy.
muzero
A clean implementation of MuZero and AlphaZero following the AlphaZero General framework. Train and Pit both algorithms against each other, and investigate reliability of learned MuZero MDP models.
wandb
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
joeryjoery's Repositories
joeryjoery/jit_env
An Agent-Environment interface with Jax transformation compatibility
joeryjoery/PyCartogram
Produce Cartograms in Python with Shapely, Geopandas and numpy.
joeryjoery/acme
A library of reinforcement learning components and agents
joeryjoery/HierQ
Hierarchical multistep and Q(λ)