jhtschultz
Follow
Stars
Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry lead…
Long Range Arena for Benchmarking Efficient Transformers
A collection of tools for reverse engineering neural networks.
Python 3.8+ toolbox for submitting jobs to Slurm
An algorithm that generalizes the paradigm of self-play reinforcement learning and search to imperfect-information games.
TFRecorder makes it easy to create TensorFlow records (TFRecords) from Pandas DataFrames and CSVs files containing images or structured data.
Fast and Easy Infinite Neural Networks in Python
Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 games
Understand and mitigate software-observable side-channels
Python library to carry out DFIR analysis on the Cloud
A generic, simple and fast implementation of Deepmind's AlphaZero algorithm.
A highly-customisable gridworld game engine with some batteries included. Make your own gridworld games to test reinforcement learning agents!
Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.
The RL Reliability Metrics library provides a set of metrics for measuring the reliability of reinforcement learning (RL) algorithms, as well as statistical tools for comparing algorithms and for c…
SEED RL: Scalable and Efficient Deep-RL with Accelerated Central Inference. Implements IMPALA and R2D2 algorithms in TF2 with SEED's architecture.
A library for easy and efficient manipulation of tensor networks.
Vim plugin for syntax-aware code formatting
High-performance TensorFlow library for quantitative finance.
Google's Engineering Practices documentation
Gin provides a lightweight configuration framework for Python
bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more