Skip to content
View kaixih's full-sized avatar
73 followers · 0 following

Achievements

Achievement: StarstruckAchievement: Pair Extraordinairex2Achievement: Arctic Code Vault ContributorAchievement: Pull Sharkx2

Achievements

Achievement: StarstruckAchievement: Pair Extraordinairex2Achievement: Arctic Code Vault ContributorAchievement: Pull Sharkx2

Block or report kaixih

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Please don't include any personal information such as legal names or email addresses. Maximum 100 characters, markdown supported. This note will be visible to only you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse

  • maxtext Public

    Forked from AI-Hypercomputer/maxtext

    A simple, performant and scalable Jax LLM!

    Python Apache License 2.0 Updated Feb 12, 2025

  • vllm Public

    Forked from vllm-project/vllm

    A high-throughput and memory-efficient inference and serving engine for LLMs

    Python Apache License 2.0 Updated Feb 12, 2025

  • jax Public

    Forked from jax-ml/jax

    Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

    Python Apache License 2.0 Updated Dec 19, 2024

  • xla Public

    Forked from openxla/xla

    A machine learning compiler for GPUs, CPUs, and ML accelerators

    C++ Apache License 2.0 Updated Nov 26, 2024

  • xla_fp8_perf_test Public

    Shell Updated Sep 25, 2024

  • praxis Public

    Forked from google/praxis
    Python Apache License 2.0 Updated Jul 31, 2024

  • paxml Public

    Forked from google/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 lead…

    Python Apache License 2.0 Updated Jul 23, 2024

  • flax Public

    Forked from google/flax

    Flax is a neural network library for JAX that is designed for flexibility.

    Python Apache License 2.0 Updated Jul 11, 2024

  • JAX-Toolbox Public

    Forked from NVIDIA/JAX-Toolbox

    JAX-Toolbox

    Python Apache License 2.0 Updated May 31, 2024

  • cudnn_frontend_test Public

    A scalable code framework for CuDNN frontend APIs

    C++ 4 2 Apache License 2.0 Updated Apr 23, 2024

  • benchmark_layer_norm Public

    HTML Updated Mar 1, 2024

  • tensorflow Public

    Forked from tensorflow/tensorflow

    An Open Source Machine Learning Framework for Everyone

    C++ Apache License 2.0 Updated Dec 16, 2023

  • atex Public

    Forked from NVIDIA/atex

    A TensorFlow Extension: GPU performance tools for TensorFlow.

    Python BSD 3-Clause "New" or "Revised" License Updated Aug 15, 2023

  • TransformerEngine Public

    Forked from NVIDIA/TransformerEngine

    A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper GPUs, to provide better performance with lower memory utilization in bot…

    Cuda Apache License 2.0 Updated Jun 8, 2023

  • kaixih.github.io Public

    JavaScript MIT License Updated Apr 30, 2023

  • JAX101 Public

    Jupyter Notebook Updated Apr 17, 2023

  • iree_playground Public

    Python Updated Mar 30, 2023

  • xla_fp8_testing Public

    Python Updated Mar 24, 2023

  • jax_101 Public

    Python Updated Mar 23, 2023

  • dl_samples Public

    Code samples to use deep learning frameworks, libraries.

    Cuda 1 MIT License Updated Dec 8, 2022

  • tf_op_graph Public

    A visualization tool to display TF-Grappler optimized op graph

    Python 12 5 MIT License Updated Aug 31, 2022

  • benchmark_gemm_fusion Public

    C++ 1 Updated Jun 27, 2022

  • depthwise_conv2d_benchmark Public

    Cuda 1 Updated May 17, 2022

  • cudnn_repro Public

    Cuda Updated Mar 9, 2022

  • cudnn_migration Public

    Migrate your cuDNN v7 legacy APIs to cuDNN v8 frontend APIs

    C++ 4 4 Updated Jan 26, 2022

  • layer_norm Public

    C++ 1 Updated Dec 20, 2021

  • MHA_EinsumDense Public

    Python Updated Nov 16, 2021

  • tmp_logs Public

    Updated May 26, 2021

  • horovod Public

    Forked from horovod/horovod

    Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

    Python 1 Other Updated Apr 23, 2021

  • slate Public

    Forked from pages-themes/slate

    Slate is a Jekyll theme for GitHub Pages

    CSS Creative Commons Zero v1.0 Universal Updated Dec 15, 2020