Repositories list

• pytorch-approx-topk

  Public
  Bucketed top-k for PyTorch using a priority queue

  pytorchkernelstopk

  Python

  •

  MIT License

  •0•2•0•0•Updated Dec 12, 2024Dec 12, 2024

• gfloat

  Public
  Generic floating-point types in Python

  Python

  •

  MIT License

  •2•12•0•1•Updated Dec 5, 2024Dec 5, 2024

• graphcore-research.github.io

  Public
  HTML

  •

  MIT License

  •0•0•0•0•Updated Dec 3, 2024Dec 3, 2024

• unit-scaling

  Public
  A library for unit scaling in PyTorch

  Jupyter Notebook

  •

  Apache License 2.0

  •8•113•8•0•Updated Nov 28, 2024Nov 28, 2024

• kg-topology-toolbox

  Public
  A Python toolbox to compute topological metrics and statistics for Knowledge Graphs

  knowledge-graphdata-analysismetrics-gathering+ 1graph-topology

  Python

  •

  MIT License

  •0•7•0•0•Updated Nov 8, 2024Nov 8, 2024

• jax-scalify

  Public
  JAX Scalify: end-to-end scaled arithmetics

  jaxlow-precisionllm+ 1fp8

  Python

  •

  Apache License 2.0

  •0•12•7•1•Updated Oct 30, 2024Oct 30, 2024

• minimol

  Public
  MiniMol is a 10M-parameters molecular fingerprinting model pre-trained on >3300 biological and quantum tasks

  Jupyter Notebook

  •

  MIT License

  •2•14•0•0•Updated Oct 7, 2024Oct 7, 2024

• mess

  Public archive
  MESS: Modern Electronic Structure Simulations

  density-functional-theoryelectronic-structurejax

  Python

  •

  MIT License

  •3•17•0•0•Updated Sep 24, 2024Sep 24, 2024

• track-and-visualize

  Public
  Track & Visualisation tool for numerics debugging

  Python

  •

  MIT License

  •0•6•2•0•Updated Sep 20, 2024Sep 20, 2024

• llm-inference-research

  Public
  An experimentation platform for LLM inference optimisation

  Jupyter Notebook

  •

  MIT License

  •1•27•0•0•Updated Sep 19, 2024Sep 19, 2024

• sparq-gpt-fast

  Public
  Fork of pytorch-labs/gpt-fast with SparQ attention and benchmarking

  Python

  •

  BSD 3-Clause "New" or "Revised" License

  •1•1•0•0•Updated Sep 19, 2024Sep 19, 2024

• sparq-llama.cpp

  Public
  LLM inference in C/C++

  C++

  •

  MIT License

  •10k•0•0•0•Updated Sep 16, 2024Sep 16, 2024

• ml_dtypes

  Public
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.

  C++

  •

  Apache License 2.0

  •29•0•0•0•Updated Sep 13, 2024Sep 13, 2024

• out-of-the-box-fp8-training

  Public
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.

  Jupyter Notebook

  •

  MIT License

  •3•39•0•0•Updated Jul 17, 2024Jul 17, 2024

• graphium-smg

  Public
  Graphium fork for Scaling Molecular GNNs project at Graphcore

  Python

  •

  Apache License 2.0

  •12•0•0•3•Updated Apr 8, 2024Apr 8, 2024

• bess-kge

  Public
  A PyTorch library for Knowledge Graph Embedding on Graphcore IPUs implementing the distribution framework BESS

  machine-learningaideep-learning+ 4knowledge-graphipukgegraphcore

  Jupyter Notebook

  •

  MIT License

  •2•3•0•0•Updated Mar 21, 2024Mar 21, 2024

• ipu-ray-lib

  Public
  Path-tracer with Neural HDRI for Graphcore IPUs.

  C++

  •

  Other

  •2•3•0•0•Updated Mar 12, 2024Mar 12, 2024

• tessellate-ipu

  Public
  TessellateIPU: low level Poplar tile programming from Python

  machine-learningaideep-learning+ 3ipujaxgraphcore

  Python

  •

  Apache License 2.0

  •0•12•4•1•Updated Mar 12, 2024Mar 12, 2024

• flash-attention-ipu

  Public
  Poplar implementation of FlashAttention for IPU

  deep-learningtransformerspytorch+ 5ipugraphcorepoplarflash-attentionflash-attention-2

  C++

  •

  MIT License

  •0•3•0•0•Updated Mar 12, 2024Mar 12, 2024

• unit-scaling-demo

  Public
  Unit Scaling demo and experimentation code

  deep-learningkeraslanguage-model+ 1low-precision-training

  Jupyter Notebook

  •

  MIT License

  •2•16•0•0•Updated Mar 12, 2024Mar 12, 2024

• pyscf-ipu

  Public
  PySCF on IPU

  dftipujax+ 2gnngraphcore

  C++

  •

  Apache License 2.0

  •2•42•30•9•Updated Mar 12, 2024Mar 12, 2024

• sparsity-benchmarks

  Public
  Benchmarking code for “PopSparse: Accelerated block sparse matrix multiplication on IPU”

  Jupyter Notebook

  •

  MIT License

  •0•2•0•0•Updated Mar 12, 2024Mar 12, 2024

• jax-experimental

  Public
  JAX for Graphcore IPU (experimental)

  aiipujax+ 3graphcoremachine-learningdeep-learning

  Python

  •

  Apache License 2.0

  •2•21•4•1•Updated Mar 12, 2024Mar 12, 2024

• notebooks

  Public
  Small, standalone, machine learning & IPU notebooks

  Jupyter Notebook

  •

  MIT License

  •0•2•0•0•Updated Feb 29, 2024Feb 29, 2024

• .github

  Public
  0•0•0•0•Updated Jan 26, 2024Jan 26, 2024

• qm1b-dataset

  Public
  one billion quantum mechanical simulations containing 9-11 heavy atoms

  Python

  •

  MIT License

  •0•4•0•0•Updated Nov 13, 2023Nov 13, 2023

• pytorch-tensor-tracker

  Public
  Flexibly track outputs and grad-outputs of torch.nn.Module.

  Python

  •

  MIT License

  •1•13•0•0•Updated Oct 6, 2023Oct 6, 2023

• jit-dynamic-lookup

  Public
  An experimental dynamic tensor slice operation, using JIT-compiled data exchanges

  C++

  •

  MIT License

  •1•5•0•0•Updated Sep 26, 2023Sep 26, 2023

• making-efficientnet-more-efficient

  Public
  Supplementary materials for: Making EfficientNet More Efficient: Exploring Batch-Independent Normalization, Group Convolutions and Reduced Resolution Training

  MIT License

  •0•0•0•0•Updated Sep 26, 2023Sep 26, 2023

• tensorflow-jax-experimental

  Public
  TensorFlow XLA backend of experimental JAX on Mk2

  C++

  •

  Apache License 2.0

  •1•1•0•1•Updated Sep 19, 2023Sep 19, 2023