yeandy
AI / ML Software Engineer and Researcher. Experienced in high performance deep learning, inference, LLMs, evaluation, and agentic systems.
Pinned Repositories
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.
autogen
A programming framework for agentic AI 🤖 PyPi: autogen-agentchat Discord: https://aka.ms/autogen-discord Office Hour: https://aka.ms/autogen-officehour
inference
Reference implementations of MLPerf™ inference benchmarks
autogen
A programming framework for agentic AI 🤖
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, ...
yeandy's Repositories
yeandy/autogen
A programming framework for agentic AI 🤖
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.