Pinned Repositories
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
analytics-zoo
Distributed Tensorflow, Keras and PyTorch on Apache Spark/Flink & Ray
autograd
Efficiently computes derivatives of numpy code.
benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
BigDL
BigDL: Distributed Deep Learning Library for Apache Spark
coding-interview-university
A complete computer science study plan to become a software engineer.
diffusers
Modification to run inference of Stable Diffusion models from HF. Will be included in the tutorial.
docker-python
Kaggle Python docker image
PyTorchXLA
Enabling PyTorch on Google TPU
tensorflow
Computation using data flow graphs for scalable machine learning
ManfeiBai's Repositories
ManfeiBai/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
ManfeiBai/analytics-zoo
Distributed Tensorflow, Keras and PyTorch on Apache Spark/Flink & Ray
ManfeiBai/autograd
Efficiently computes derivatives of numpy code.
ManfeiBai/benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
ManfeiBai/BigDL
BigDL: Distributed Deep Learning Library for Apache Spark
ManfeiBai/diffusers
Modification to run inference of Stable Diffusion models from HF. Will be included in the tutorial.
ManfeiBai/EDSR-PyTorch
PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)
ManfeiBai/flash-attention
Fast and memory-efficient exact attention
ManfeiBai/flash-attention-jax
Implementation of Flash Attention in Jax
ManfeiBai/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
ManfeiBai/PyTorchXLA
Enabling PyTorch on Google TPU
ManfeiBai/tensorflow
Computation using data flow graphs for scalable machine learning
ManfeiBai/jetstream-pytorch
PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
ManfeiBai/llama
Inference code for LLaMA models
ManfeiBai/maxdiffusion
ManfeiBai/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.
ManfeiBai/ml-testing-accelerators
Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
ManfeiBai/NEU-Cluster
Unsupervised classification of the North Eastern University Steel Surface Defects Database
ManfeiBai/OpenXLA
A community-driven and modular open source compiler for ML.
ManfeiBai/optimum-nvidia
ManfeiBai/optimum-tpu-1
Google TPU optimizations for transformers models
ManfeiBai/pdfs
Technically-oriented PDF Collection (Papers, Specs, Decks, Manuals, etc)
ManfeiBai/pydata-book
Materials and IPython notebooks for "Python for Data Analysis" by Wes McKinney, published by O'Reilly Media
ManfeiBai/Python
All Algorithms implemented in Python
ManfeiBai/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
ManfeiBai/run_torchbench
This repo contains a list of stand alone scripts for running torchbench models one by one.
ManfeiBai/spark
Apache Spark - A unified analytics engine for large-scale data processing
ManfeiBai/torchvision
Datasets, Transforms and Models specific to Computer Vision
ManfeiBai/transformers
🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.
ManfeiBai/xgboost
Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow