Skip to content
@pytorch-tpu

pytorch-tpu

Popular repositories Loading

  1. examples examples Public

    This repository contains example code to build models on TPUs

    Shell 30 11

  2. fairseq fairseq Public

    Forked from facebookresearch/fairseq

    Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

    Python 22 7

  3. llama llama Public

    Forked from meta-llama/llama

    Inference code for LLaMA models

    Python 20 10

  4. transformers transformers Public

    Forked from huggingface/transformers

    🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.

    Python 13 9

  5. ray-llama ray-llama Public

    Example for running Llama 2 on Ray with TPUs

    Python 10 3

  6. stable-diffusion stable-diffusion Public

    Forked from CompVis/stable-diffusion

    A latent text-to-image diffusion model

    Jupyter Notebook 7

Repositories

Showing 10 of 12 repositories
  • diffusers Public

    Modification to run inference of Stable Diffusion models from HF. Will be included in the tutorial.

    pytorch-tpu/diffusers’s past year of commit activity
    Python 2 Apache-2.0 2 0 1 Updated Nov 19, 2024
  • run_torchbench Public

    This repo contains a list of stand alone scripts for running torchbench models one by one.

    pytorch-tpu/run_torchbench’s past year of commit activity
    Python 0 BSD-3-Clause 2 0 1 Updated Nov 19, 2024
  • transformers Public Forked from huggingface/transformers

    🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.

    pytorch-tpu/transformers’s past year of commit activity
    Python 13 Apache-2.0 27,474 2 5 Updated Nov 10, 2024
  • accelerate Public Forked from huggingface/accelerate

    🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

    pytorch-tpu/accelerate’s past year of commit activity
    Python 0 Apache-2.0 981 0 0 Updated Oct 17, 2024
  • llama Public Forked from meta-llama/llama

    Inference code for LLaMA models

    pytorch-tpu/llama’s past year of commit activity
    Python 20 9,725 4 8 Updated May 30, 2024
  • llm-foundry Public Forked from mosaicml/llm-foundry

    LLM training code for Databricks foundation models

    pytorch-tpu/llm-foundry’s past year of commit activity
    Python 0 Apache-2.0 536 0 0 Updated May 30, 2024
  • composer Public Forked from mosaicml/composer

    Supercharge Your Model Training

    pytorch-tpu/composer’s past year of commit activity
    Python 0 Apache-2.0 438 0 0 Updated May 29, 2024
  • ray-llama Public

    Example for running Llama 2 on Ray with TPUs

    pytorch-tpu/ray-llama’s past year of commit activity
    Python 10 Apache-2.0 3 1 0 Updated Nov 28, 2023
  • stable-diffusion Public Forked from CompVis/stable-diffusion

    A latent text-to-image diffusion model

    pytorch-tpu/stable-diffusion’s past year of commit activity
    Jupyter Notebook 7 10,329 0 0 Updated Sep 11, 2023
  • training Public Forked from mlcommons/training

    Reference implementations of MLPerf™ training benchmarks

    pytorch-tpu/training’s past year of commit activity
    Python 3 564 0 0 Updated Sep 8, 2023
View all repositories

People

This organization has no public members. You must be a member to see who’s a part of this organization.

Top languages

Loading…

Most used topics

Loading…