fl0wbar

fl0wbar's Stars

• tensorflow/tensorflow

  An Open Source Machine Learning Framework for Everyone

  Language:C++189k7.5k40.3k74.6k

• huggingface/transformers

  🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

  Language:Python141k1.1k16.9k28.3k

• pytorch/pytorch

  Tensors and Dynamic neural networks in Python with strong GPU acceleration

  Language:Python88k1.8k49.6k23.6k

• huggingface/pytorch-image-models

  The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

  Language:Python33.5k3199594.9k

• jax-ml/jax

  Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

  Language:Python31.7k3336k2.9k

• Lightning-AI/pytorch-lightning

  Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

  Language:Python29.1k2517.3k3.5k

• huggingface/diffusers

  🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.

  Language:Python28.1k2124.6k5.8k

• pyg-team/pytorch_geometric

  Graph Neural Network Library for PyTorch

  Language:Python22k2543.6k3.8k

• huggingface/datasets

  🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools

  Language:Python19.8k2763k2.8k

• huggingface/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

  Language:Python8.5k961.7k1k

• crowsonkb/k-diffusion

  Karras et al. (2022) diffusion models for PyTorch

  Language:Python2.4k4267383

• KellerJordan/modded-nanogpt

  NanoGPT (124M) in 3 minutes

  Language:Python2.4k3931265

• patrick-kidger/diffrax

  Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

  Language:Python1.6k20384144

• tensorflow/compression

  Data compression in TensorFlow

  Language:Python87444102254

• KellerJordan/cifar10-airbench

  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds

  Language:Python2174214

• HomebrewML/HeavyBall

  Efficient optimizers

  Language:Python182103313

• lixilinx/psgd_torch

  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation preconditioner and more)

  Language:Python1695310

• paradigms-of-intelligence/cubff

  Language:C++1184129

• znah/zff

  8-bit computational substrates

  Language:JavaScript39216

• evanatyourservice/psgd_jax

  Implementation of PSGD optimizer in JAX

  Language:Python28402

• lixilinx/Fully-Trainable-SSM

  A fully trainable state space model (SSM)

  Language:Python3