A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its variants as the primary backbone with support for ImageNet train…

Gemma open-weight LLM library, from Google DeepMind

Octo is a transformer-based robot policy trained on a diverse mix of 800k robot trajectories.

dataloading is my passion

Nintendo Switch Teleoperation and ACT autonomy on ALOHA for Bigym benchmark tasks

Modern columnar data format for ML implemented in Rust. Convert from parquet in 2 lines of code for 100x faster random access, vector index, and data versioning. Compatible with Pandas, DuckDB, Pol…

Create UIs for your machine learning model in Python in 3 minutes

Optax is a gradient processing and optimization library for JAX.

A user-friendly launcher for Bazel.

Repository for tutorial sessions at EEML2020

TF-Agents is a library for Reinforcement Learning in TensorFlow

Use AWS RoboMaker and demonstrate running a simulation which trains a reinforcement learning (RL) model to drive a car around a track

Swift for TensorFlow Deep Learning Library

Stores documents used by the TensorFlow developer community

Swift for TensorFlow Project Home Page

OpenAI gym environment for donkeycar simulator

Imitation Learning Homework 1

A TensorFlow implementation of Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architectures.

Taskonomy: Disentangling Task Transfer Learning

Wasserstein Auto-Encoders

Simple Reinforcement learning tutorials

[EXPERIMENTAL] This repo includes deployment instructions for running HDFS/Spark inside docker containers. Also includes spark-notebook and HDFS FileBrowser.

rllab is a framework for developing and evaluating reinforcement learning algorithms, fully compatible with OpenAI Gym.