Alternatives and similar repositories for jaxlib-feedstock

Users that are interested in jaxlib-feedstock are comparing it to the libraries listed below

• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆45Updated 5 years ago
• geomstats / applications
  Examples of more involved applications using Geomstats
  ☆33Updated 4 years ago
• pierresegonne / VINF
  Repository for DTU Special Course, focusing on Variational Inference using Normalizing Flows (VINF). Supervised by Michael Riis Andersen
  ☆25Updated 5 years ago
• conda-forge / pytorch-cpu-feedstock
  A conda-smithy repository for pytorch-cpu.
  ☆30Updated last week
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 4 years ago
• IndustAI / learning-group-structure
  Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"
  ☆15Updated 2 years ago
• probml / JSL
  Jax SSM Library
  ☆48Updated 3 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆40Updated last year
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• PredictiveIntelligenceLab / JAX-BO
  ☆30Updated 3 years ago
• toshas / torch-householder
  Efficient Householder Transformation in PyTorch
  ☆66Updated 4 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• sscardapane / hydra-tutorial
  Companion code for a tutorial on using Hydra.
  ☆30Updated 4 years ago
• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆69Updated 2 months ago
• ghliu / snopt
  Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight
  ☆55Updated 4 years ago
• conda-forge / nvcc-feedstock
  A conda-smithy repository for nvcc.
  ☆13Updated 10 months ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆28Updated 4 years ago
• google-deepmind / tf2jax
  ☆118Updated 3 weeks ago
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 5 years ago
• borchero / pyblaze
  Large-Scale Machine and Deep Learning in PyTorch.
  ☆21Updated 4 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆69Updated 7 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆58Updated 4 months ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆117Updated 3 years ago
• google-research / gpax
  ☆31Updated last year
• patrick-kidger / FasterNeuralDiffEq
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆88Updated 3 years ago
• CMA-ES / pycomocma
  A Python implementation of COMO-CMA-ES, a non-elitist multiobjective Evolution Strategy
  ☆15Updated 2 months ago
• juliagusak / neural-ode-metasolver
  Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561
  ☆25Updated 4 years ago
• danieljtait / ladax
  Layered distributions using FLAX/JAX
  ☆10Updated 4 years ago
• mhlr / awesome-jax
  List of awesome JAX resources
  ☆13Updated 2 years ago
• cunningham-lab / cb_and_cc
  ☆37Updated 3 years ago