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.
  ☆44Updated 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
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆57Updated last month
• google-research / gpax
  ☆30Updated 11 months ago
• ml-jku / quam
  Quantification of Uncertainty with Adversarial Models
  ☆30Updated 2 years ago
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 3 years ago
• catid / dataloader
  High-performance tokenized language data-loader for Python C++ extension
  ☆13Updated last year
• geomstats / applications
  Examples of more involved applications using Geomstats
  ☆32Updated 4 years ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆50Updated 3 years ago
• juliusberner / deep_kolmogorov
  Numerically Solving Parametric Families of High-Dimensional Kolmogorov Partial Differential Equations via Deep Learning (NeurIPS 2020)
  ☆22Updated 3 years ago
• ghliu / snopt
  Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight
  ☆53Updated 3 years ago
• TorchUQ / torchuq
  A library for uncertainty quantification based on PyTorch
  ☆122Updated 3 years ago
• probml / JSL
  Jax SSM Library
  ☆49Updated 2 years ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆57Updated 3 months 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
• attentionneuron / attentionneuron.github.io
  ☆26Updated 2 years ago
• mhlr / awesome-jax
  List of awesome JAX resources
  ☆13Updated 2 years ago
• toshas / torch-householder
  Efficient Householder Transformation in PyTorch
  ☆66Updated 4 years ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆28Updated 3 years ago
• PredictiveIntelligenceLab / JAX-BO
  ☆30Updated 3 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆112Updated 3 years ago
• patrick-kidger / patdb
  A snappy + easy + pretty TUI debugger for Python.
  ☆64Updated last month
• Physics-aware-AI / DiffCoSim
  By introducing a differentiable contact model, DiffCoSim extends the applicability of Lagrangian/Hamiltonian-inspired neural networks to …
  ☆36Updated 2 years ago
• ejmichaud / precision-ml
  ☆13Updated 2 years ago
• ronaldiscool / VAETutorial
  ☆17Updated 5 years ago
• oxcsml / lie-transformer
  LieTransformer: Equivariant Self-Attention for Lie Groups
  ☆72Updated last year
• AlexandraVolokhova / stochasticity_in_neural_ode
  "Stochasticity in Neural ODEs: An Empirical Study". Experiments from the paper
  ☆13Updated 5 years ago
• nestordemeure / flaxOptimizers
  A collection of optimizers, some arcane others well known, for Flax.
  ☆29Updated 4 years ago
• cgarciae / einop
  ☆60Updated 3 years ago