Skip to content

Instantly share code, notes, and snippets.

@josephrocca
josephrocca / deepspeed-compression.ipynb
Created August 5, 2022 09:56
DeepSpeed extreme compression
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@josephrocca
josephrocca / minimal-dalle-mini-inference-and-tflite-conversion-produces-zero-byte-tflite-model.ipynb
Created August 3, 2022 19:15
Minimal dalle-mini inference and tflite conversion (produces zero-byte tflite model)
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
joe@6610f9ed85bc:/src/packages/jax$ python -m pyodide_build buildpkg meta.yaml
[2022-07-07 05:32:43] Building package jax...
Hit:1 http://deb.debian.org/debian buster InRelease
Hit:2 http://security.debian.org/debian-security buster/updates InRelease
Hit:3 https://dl.google.com/linux/chrome/deb stable InRelease
Hit:4 http://deb.debian.org/debian buster-updates InRelease
Reading package lists...
Reading package lists...
Building dependency tree...
Reading state information...
@josephrocca
josephrocca / meta.yaml
Last active July 7, 2022 04:17
meta.yaml for JAX Pyodide port
package:
name: jax
version: 0.3.14 # not the real version - actual src is: `git clone --branch patch-4 --depth 1 https://github.com/josephrocca/jax`
source:
path: /src/jax
build:
script: |
sudo apt-get update
sudo apt-get install -y python3-dev
pip install numpy wheel
@josephrocca
josephrocca / swinir-demo-on-real-world-image-sr.ipynb
Created July 6, 2022 16:11
SwinIR PyTorch model init time test.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@josephrocca
josephrocca / swinit-40k-nodes-in-model-init-time-test.ipynb
Created July 6, 2022 15:30
SwinIT ~40k nodes in model - init time test.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@josephrocca
josephrocca / minimal-dalle-mini-inference-and-tflite-onnx-tfjs-conversion.ipynb
Created July 6, 2022 13:23
Minimal dalle-mini inference and tflite/onnx/tfjs conversion
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@josephrocca
josephrocca / tflite-dalle-mini-conversion-separated-encoder-and-decoder.ipynb
Created July 2, 2022 16:58
tflite dalle mini conversion - separated encoder and decoder.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@josephrocca
josephrocca / jax-flax-transformations-cannot-be-mixed.ipynb
Created July 2, 2022 16:00
jax-flax-transformations-cannot-be-mixed.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@josephrocca
josephrocca / jax-flax-transformations-cannot-be-mixed.ipynb
Created July 2, 2022 15:55
jax-flax-transformations-cannot-be-mixed.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.