google-deepmind / s6

Related projects: ⓘ

• AMYPAD / CuVec
  Unifying Python/C++/CUDA memory: Python buffered array ↔️ `std::vector` ↔️ CUDA managed memory
  ☆80Updated 3 weeks ago
• a-rahimi / python-checkpointing2
  Checkpoint the state of Python programs using Pythonic setjmp and longjmp
  ☆65Updated 3 years ago
• bwasti / better_bindings
  Better bindings for Python
  ☆17Updated last year
• facebookresearch / loop_tool
  A thin, highly portable toolkit for efficiently compiling dense loop-based computation.
  ☆147Updated last year
• salykova / matmul.c
  Fast multi-threaded matrix multiplication in C
  ☆164Updated 3 weeks ago
• sradc / SmallPebble
  Minimal deep learning library written from scratch in Python, using NumPy/CuPy.
  ☆116Updated 2 years ago
• APrioriInvestments / typed_python
  An llvm-based framework for generating and calling into high-performance native code from Python.
  ☆197Updated 7 months ago
• google / etils
  Collection of eclectic utils for python.
  ☆182Updated this week
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆136Updated this week
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆194Updated this week
• breandan / picograd
  🦠 AD in less than 20 lines
  ☆48Updated 3 years ago
• minotaur-toolkit / minotaur
  A description of Minotaur can be found in https://arxiv.org/abs/2306.00229.
  ☆86Updated last month
• chalk-diagrams / chalk
  A declarative drawing API in Python
  ☆275Updated 3 weeks ago
• pytorch / expecttest
  ☆71Updated last month
• a1k0n / a1gpt
  throwaway GPT inference
  ☆139Updated 3 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆94Updated this week
• hpyproject / hpyproject.org
  Use Nikola to build hpypjoject.org
  ☆23Updated 5 months ago
• Pylir / Pylir
  An optimizing ahead-of-time Python Compiler
  ☆155Updated 3 months ago
• erezsh / runtype
  Utilities for run-time type validation and multiple dispatch
  ☆161Updated last week
• joennlae / halutmatmul
  Hashed Lookup Table based Matrix Multiplication (halutmatmul) - Stella Nera accelerator
  ☆205Updated 9 months ago
• AZHenley / superoptimizer
  A toy superoptimizer for a limited assembly language.
  ☆56Updated last year
• Quansight-Labs / free-threaded-compatibility
  A central repository to keep track of the status of work on and support for free-threaded CPython (see PEP 703), with a focus on the scie…
  ☆123Updated last week
• IntelPython / DPEP
  Data Parallel Extensions for Python*
  ☆32Updated last month
• mlecauchois / micrograd-cuda
  ☆230Updated 6 months ago
• DiscoGrad / DiscoGrad
  DiscoGrad - automatically differentiate across conditional branches in C++ programs
  ☆202Updated last week
• atom-moyer / getpy
  A Vectorized Python Dict/Set
  ☆115Updated last year
• google-deepmind / tensor_annotations
  Annotating tensor shapes using Python types
  ☆158Updated last year
• glassroom / heinsen_sequence
  Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)
  ☆74Updated 8 months ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆84Updated 7 months ago
• google / tree-math
  Mathematical operations for JAX pytrees
  ☆181Updated 4 months ago