yeandy

JetStream

JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).

maxdiffusion

maxtext

A simple, performant and scalable Jax LLM!

ml-auto-solutions

A simplified and automated orchestration workflow to perform ML end-to-end (E2E) model tests and benchmarking on Cloud VMs across different frameworks.

inference

Reference implementations of MLPerf™ inference benchmarks

beam

Apache Beam is a unified programming model for Batch and Streaming

datasets

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

DeepLearningExamples

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

yeandy/beam

Apache Beam is a unified programming model for Batch and Streaming

yeandy/datasets

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

yeandy/DeepLearningExamples

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

yeandy/diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

yeandy/edgebundleR

Circle plot with bundled edges

yeandy/FasterTransformer

Transformer related optimization, including BERT, GPT

yeandy/pandas

Flexible and powerful data analysis / manipulation library for Python, providing labeled data structures similar to R data.frame objects, statistical functions, and much more

yeandy/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

yeandy/gradio

Create UIs for prototyping your machine learning model in 3 minutes

yeandy/inference

Reference implementations of MLPerf™ inference benchmarks

yeandy/paxml

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 leading model flop utilization rates.

yeandy/praxis

yeandy/saxml

yeandy/stable-diffusion-jax

yeandy/tfx-bsl

Common code for TFX

yeandy/transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.