lucadiliello / pytorch-apple-silicon-benchmarks
Performance of PyTorch on Apple Silicon
☆47Updated last year
Alternatives and similar repositories for pytorch-apple-silicon-benchmarks:
Users that are interested in pytorch-apple-silicon-benchmarks are comparing it to the libraries listed below
- ☆15Updated 2 years ago
- 3X speedup over Apple’s TensorFlow plugin by using Apache TVM on M1☆135Updated 2 years ago
- MLX support for the Open Neural Network Exchange (ONNX)☆44Updated 11 months ago
- FlashAttention (Metal Port)☆436Updated 4 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆107Updated 3 weeks ago
- PyTorch's full-scratch build and install for Apple Silicon☆30Updated last year
- A conda-smithy repository for nvcc.☆12Updated 3 weeks ago
- TensorFlow Metal Backend on Apple Silicon Experiments (just for fun)☆277Updated 3 years ago
- Nod.ai 🦈 version of 👻 . You probably want to start at https://github.com/nod-ai/shark for the product and the upstream IREE repository …☆106Updated last month
- Graph Neural Network library made for Apple Silicon☆183Updated 4 months ago
- In-depth code associated with my Medium blog post, "How to Load PyTorch Models 340 Times Faster with Ray"☆26Updated 2 years ago
- Benchmark of Apple MLX operations on all Apple Silicon chips (GPU, CPU) + MPS and CUDA.☆153Updated 3 months ago
- Contrastive Language-Image Pretraining☆38Updated 7 months ago
- LibTorch builds for the M1 Macs☆53Updated 3 months ago
- AdamW optimizer for bfloat16 models in pytorch 🔥.☆31Updated 8 months ago
- Code for "Training-free Graph Neural Networks and the Power of Labels as Features" (TMLR 2024)☆55Updated 6 months ago
- mlx image models for Apple Silicon machines☆71Updated 2 months ago
- Fast and simple stream processing of files in tar files, useful for deep learning, big data, and many other applications.☆127Updated last year
- New operators for the ReferenceEvaluator, new kernels for onnxruntime, CPU, CUDA☆32Updated 4 months ago
- ☆50Updated 3 years ago
- Triton Server Component for lightning.ai☆14Updated 2 years ago
- GPT-J 6B inference on TensorRT with INT-8 precision☆11Updated last year
- ☆54Updated last year
- A logging tool for deep learning.☆53Updated 2 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆74Updated last year