36 results found Sort:

vllm-project
vllm vllm-project
4.5k
29.7k
apache-2.0
242
A high-throughput and memory-efficient inference and serving engine for LLMs
amd gpt hpu llm tpu xpu cuda rocm llama mlops llmops pytorch trainium inference inferentia llm-serving transformer model-serving
Created 2023-02-09
3,290 commits to main branch, last one 7 hours ago
tensorflow
tensor2tensor tensorflow
3.5k
15.5k
apache-2.0
466
Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.
tpu deep-learning machine-learning machine-translation reinforcement-learning
This repository has been archived (exclude archived)
Created 2017-06-15
4,379 commits to master branch, last one about a year ago
skypilot-org
skypilot skypilot-org
502
6.8k
apache-2.0
71
SkyPilot: Run AI and batch jobs on any infra (Kubernetes or 12+ clouds). Get unified execution, cost savings, and high GPU availability via a simple interface.
gpu tpu finops job-queue multicloud llm-serving ml-platform data-science llm-training deep-learning job-scheduler spot-instances cloud-computing cost-management cloud-management machine-learning cost-optimization ml-infrastructure distributed-training hyperparameter-tuning
Created 2021-08-11
2,082 commits to master branch, last one 7 hours ago
tensorflow
adanet tensorflow
527
3.5k
apache-2.0
172
Fast and flexible AutoML with learning guarantees.
gpu tpu automl python ensemble tensorflow deep-learning learning-theory machine-learning distributed-training neural-architecture-search
Created 2018-06-28
440 commits to master branch, last one 3 years ago
hollance
neural-engine hollance
77
2.1k
mit
77
Everything we actually know about the Apple Neural Engine (ANE)
ane ios tpu coreml iphone neural-engine neural-network
Created 2020-04-15
77 commits to master branch, last one about a month ago
imcaspar
gpt2-ml imcaspar
334
1.7k
apache-2.0
38
GPT2 for Multiple Languages, including pretrained models. GPT2 多语言支持, 15亿参数中文预训练模型
nlp tpu bert colab gpt-2 chinese tensorflow text-generation pretrained-models
Created 2019-11-05
98 commits to master branch, last one 3 years ago
PygmalionAI
aphrodite-engine PygmalionAI
121
1.1k
agpl-3.0
14
Large-scale LLM inference engine
tpu cuda lora rocm intel api-rest inferentia inference-engine machine-learning speculative-decoding
Created 2023-06-23
801 commits to main branch, last one 2 days ago
ayaka14732
tpu-starter ayaka14732
27
491
cc-by-4.0
8
Everything you want to know about Google Cloud TPU
gcp jax tpu cloud-tpu deep-learning machine-learning google-cloud-platform
Created 2022-02-28
66 commits to main branch, last one 3 months ago
chrisbutner
ChessCoach chrisbutner
20
463
gpl-3.0
8
Neural network-based chess engine capable of natural language commentary
cpp gpu nlg tpu uci chess keras alphazero tensorflow lichess-bot chess-engine commentary-generation
Created 2020-03-14
505 commits to main branch, last one about a year ago
tumaer
JAXFLUIDS tumaer
54
328
other
16
Differentiable Fluid Dynamics Package
cfd gpu hpc jax tpu cuda jaxfluids turbulence deep-learning gpu-computing fluid-dynamics high-performance machine-learning multi-phase-flows compressible-flows automatic-differentiation computational-fluid-dynamics
Created 2022-03-21
29 commits to main branch, last one about a month ago
magic-blue-smoke
Dual-Edge-TPU-Adapter magic-blue-smoke
4
305
unknown
54
Dual Edge TPU Adapter to use it on a system with single PCIe port on m.2 A/B/E/M slot
m2 tpu coral edge-ai edgetpu edge-tpu coral-tpu m2-module pcie-card home-assistant pcie-interface tpu-benchmarks tensorflow-lite tpu-acceleration
Created 2021-06-26
24 commits to main branch, last one about a year ago
embedeep
Free-TPU embedeep
58
238
mit
10
Free TPU for FPGA with compiler supporting Pytorch/Caffe/Darknet/NCNN. An AI processor for using Xilinx FPGA to solve image classification, detection, and segmentation problem.
npu rnn tpu fpga free lstm zynq caffe darknet pytorch hardware npu-compiler deep-learning cnn-accelerator
Created 2018-12-25
44 commits to master branch, last one about a year ago
AI-Hypercomputer
JetStream AI-Hypercomputer
31
227
apache-2.0
17
JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).
gpt gpu jax llm tpu gemma llama mlops llama2 llmops pytorch inference transformer llm-inference model-serving large-language-models
Created 2024-03-01
121 commits to main branch, last one 2 days ago
Kohulan
DECIMER-Image_Transformer Kohulan
52
203
mit
7
DECIMER: Deep Learning for Chemical Image Recognition using Efficient-Net V2 + Transformer
tpu python decimer tensorflow transformers deep-learning image-data-mining chemical-image-recognition
Created 2020-09-07
317 commits to master branch, last one 2 days ago
robotperf
benchmarks robotperf
16
154
apache-2.0
10
Benchmarking suite to evaluate 🤖 robotics computing performance. Vendor-neutral. ⚪Grey-box and ⚫Black-box approaches.
cpu gpu tpu fpga ros2 robotics performance acceleration benchmarking
Created 2022-12-11
1,420 commits to main branch, last one 3 months ago
nyx-ai
stylegan2-flax-tpu nyx-ai
11
130
unknown
3
🖼 Training StyleGAN2 on TPUs in JAX
jax tpu generative-model artificial-intelligence generative-adversarial-network
Created 2022-07-15
23 commits to main branch, last one 2 years ago
rwightman
efficientnet-jax rwightman
14
127
apache-2.0
6
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
jax tpu flax objax mixnet flax-linen mobilenetv2 mobilenetv3 efficientnet
Created 2020-09-12
50 commits to master branch, last one 10 months ago
HomebrewNLP
revlib HomebrewNLP
6
124
bsd-2-clause
4
Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
tpu xla revnet pytorch deepspeed momentumnet deep-learning
Created 2021-08-17
97 commits to master branch, last one 2 years ago
embedeep
FREE-TPU-V3plus-for-FPGA embedeep
13
108
mit
6
FREE TPU V3plus for FPGA is the free version of a commercial AI processor (EEP-TPU) for Deep Learning EDGE Inference
ai npu tpu fpga lstm zynq caffe pytorch accelerator ai-compiler transformer ai-processor npu-compiler tpu-acceleration transformer-accelerator
Created 2023-05-06
14 commits to main branch, last one about a year ago
sayakpaul
FunMatch-Distillation sayakpaul
8
87
unknown
4
TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.
tpu keras vision bit-resnet tensorflow transfer-learning image-classification knowledge-distillation
Created 2021-07-24
19 commits to main branch, last one 3 years ago
wmcnally
evopose2d wmcnally
10
83
mit
2
EvoPose2D is a two-stage human pose estimation model that was designed using neuroevolution. It achieves state-of-the-art accuracy on COCO.
tpu tensorflow tensorflow2 deep-learning pose-estimation human-pose-estimation
Created 2020-10-29
198 commits to master branch, last one 3 years ago
AI-Hypercomputer
xpk AI-Hypercomputer
23
80
apache-2.0
21
xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerators such as TPUs and GPUs on GKE.
gke tpu gcloud
Created 2023-06-12
162 commits to main branch, last one 2 days ago
captain-pool
GSOC captain-pool
22
68
mit
2
Repository for Google Summer of Code 2019 https://summerofcode.withgoogle.com/projects/#4662790671826944
tpu onnx keras tf-hub tensorflow tensorflow-2 super-resolution googlesummerofcode tensorflow-2-sample tensorflow-datasets enhanced-super-resolution
Created 2019-06-09
471 commits to master branch, last one 3 years ago
rickiepark
deep-learning-with-python-2nd rickiepark
80
64
mit
2
<케라스 창시자에게 배우는 딥러닝 2판> 도서의 코드 저장소
cnn gan rnn tpu keras multi-gpu tensorflow keras-tuner time-series transformer deep-learning neural-network mixed-precision text-generation image-augmentation image-segmentation machine-translation text-classification image-classification image-style-transfer
Created 2022-04-11
106 commits to main branch, last one 8 months ago
GoogleCloudPlatform
ml-testing-accelerators GoogleCloudPlatform
59
64
apache-2.0
31
Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
gpu tpu machine-learning testing-accelerators
Created 2020-03-04
1,044 commits to master branch, last one 2 months ago
gsarti
t5-flax-gcp gsarti
6
58
apache-2.0
3
Tutorial to pretrain & fine-tune a 🤗 Flax T5 model on a TPUv3-8 with GCP
t5 gcp tpu tpu-vm seq2seq huggingface transformers text-to-text-transfer-transformer
Created 2021-09-05
6 commits to main branch, last one 2 years ago
andreped
GradientAccumulator andreped
11
51
mit
4
:dart: Accumulated Gradients for TensorFlow 2
gpu tf2 tpu keras float16 multi-gpu batch-size tensorflow huggingface tensorflow2 deep-learning hacktoberfest mixed-precision memory-constraints distributed-training accumulated-gradients gradient-accumulation adaptive-gradient-clipping accumulated-batch-normalization
Created 2022-05-31
698 commits to main branch, last one 9 months ago
instadeepai
sebulba instadeepai
3
47
apache-2.0
5
🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
ai hpc jax ppo tpu sebulba podracer deep-learning machine-learning reinforcement-learning
Created 2023-09-04
3 commits to main branch, last one about a year ago
HomebrewNLP
Olmax HomebrewNLP
6
46
bsd-2-clause
6
HomebrewNLP in JAX flavour for maintable TPU-Training
jax tpu
Created 2021-07-18
1,808 commits to main branch, last one about a year ago
Beomi
Gemma-EasyLM Beomi
11
45
apache-2.0
3
Train GEMMA on TPU/GPU! (Codebase for training Gemma-Ko Series)
jax tpu flax gemma easylm huggingface transformers language-model
Created 2024-03-02
224 commits to main branch, last one 8 months ago