Alternatives and similar repositories for jaxlib-feedstock

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

• pierresegonne / VINF
  Repository for DTU Special Course, focusing on Variational Inference using Normalizing Flows (VINF). Supervised by Michael Riis Andersen
  ☆26Updated 5 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆45Updated 5 years ago
• sscardapane / hydra-tutorial
  Companion code for a tutorial on using Hydra.
  ☆30Updated 4 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆58Updated 5 months ago
• cunningham-lab / cb_and_cc
  ☆37Updated 3 years ago
• geomstats / applications
  Examples of more involved applications using Geomstats
  ☆33Updated 4 years ago
• ml-jku / quam
  Quantification of Uncertainty with Adversarial Models
  ☆29Updated 2 years ago
• conda-forge / pytorch-cpu-feedstock
  A conda-smithy repository for pytorch-cpu.
  ☆30Updated this week
• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆70Updated 3 months ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• ghliu / snopt
  Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight
  ☆55Updated 4 years ago
• borchero / pyblaze
  Large-Scale Machine and Deep Learning in PyTorch.
  ☆21Updated 4 years ago
• probml / JSL
  Jax SSM Library
  ☆48Updated 3 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆41Updated last year
• toshas / torch-householder
  Efficient Householder Transformation in PyTorch
  ☆69Updated 4 years ago
• IndustAI / learning-group-structure
  Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"
  ☆15Updated 3 years ago
• google-deepmind / tf2jax
  ☆118Updated last month
• conda-forge / nvcc-feedstock
  A conda-smithy repository for nvcc.
  ☆13Updated 11 months ago
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 4 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆71Updated 3 weeks ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆28Updated 4 years ago
• TorchUQ / torchuq
  A library for uncertainty quantification based on PyTorch
  ☆121Updated 4 years ago
• patrick-kidger / FasterNeuralDiffEq
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆88Updated 3 years ago
• attentionneuron / attentionneuron.github.io
  ☆26Updated 3 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆118Updated 3 years ago
• catid / dataloader
  High-performance tokenized language data-loader for Python C++ extension
  ☆14Updated last year
• DiffEqML / tutorials
  ☆11Updated 3 years ago
• google-deepmind / neural_testbed
  ☆194Updated this week
• google / deluca
  Performant, differentiable reinforcement learning
  ☆125Updated 5 months ago
• PredictiveIntelligenceLab / JAX-BO
  ☆30Updated 3 years ago