j3soon / arxiv-utils

Related projects ⓘ

Alternatives and complementary repositories for arxiv-utils

• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆322Updated last year
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆328Updated last month
• ayaka14732 / jax-smi
  JAX Synergistic Memory Inspector
  ☆164Updated 4 months ago
• amitness / open-in-overleaf
  Access latex source of any arxiv.org paper directly on overleaf
  ☆149Updated 5 months ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆151Updated 2 years ago
• BlackHC / toma
  Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory
  ☆428Updated 2 months ago
• j3soon / tbparse
  Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow
  ☆181Updated 3 months ago
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆321Updated last week
• google-research / jaxpruner
  ☆207Updated 7 months ago
• kach / gradient-descent-the-ultimate-optimizer
  Code for our NeurIPS 2022 paper
  ☆363Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆201Updated this week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆344Updated this week
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆228Updated last year
• davidstutz / arxiv-submission-sanitizer-flattener
  Simple Python scripts to clean up and flatten ArXiv LaTeX submissions.
  ☆63Updated 2 years ago
• srush / do-we-need-attention
  ☆161Updated last year
• google / learned_optimization
  ☆751Updated last week
• xwying / torchshow
  Visualize PyTorch tensors with a single line of code.
  ☆647Updated 11 months ago
• Felix-Petersen / algovision
  Differentiable Algorithms and Algorithmic Supervision.
  ☆105Updated last year
• google / flaxformer
  ☆334Updated 7 months ago
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆303Updated this week
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆152Updated this week
• patrick-kidger / jaxtyping
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,227Updated this week
• ashleve / rootutils
  A simple python package to solve all your problems with pythonpath, work dir, file paths, module imports and environment variables.
  ☆142Updated last year
• google / grain
  ☆306Updated this week
• vict0rsch / PaperMemory
  Your browser's reference manager: automatic paper detection (Arxiv, OpenReview & more), publication venue matching and code repository di…
  ☆503Updated this week
• facebookresearch / FFCV-SSL
  FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.
  ☆203Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆127Updated 3 weeks ago
• klieret / wandb-offline-sync-hook
  A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!
  ☆56Updated 2 weeks ago
• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆238Updated last year
• google / ml_collections
  ML Collections is a library of Python Collections designed for ML use cases.
  ☆897Updated this week