Alternatives and similar repositories for jaxlib-feedstock:

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

• ejmichaud / precision-ml
  ☆13Updated 2 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆52Updated last week
• pierresegonne / VINF
  Repository for DTU Special Course, focusing on Variational Inference using Normalizing Flows (VINF). Supervised by Michael Riis Andersen
  ☆25Updated 4 years ago
• google-deepmind / understanding_deep_generative_models_with_generalized_empirical_likelihood
  ☆34Updated last year
• uvadlc / uvadlc_practicals_2021
  Repository for the code assignment of the Deep Learning 1 course, Fall 2021 edition
  ☆10Updated 2 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year
• gcucurull / jax-gat
  JAX implementation of Graph Attention Networks
  ☆13Updated 3 years ago
• facebookresearch / ssorl
  Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories
  ☆42Updated last year
• 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
  ☆49Updated 2 years ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 3 years ago
• google-research / gpax
  ☆30Updated 7 months ago
• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆42Updated last year
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆43Updated 4 years ago
• catid / dataloader
  High-performance tokenized language data-loader for Python C++ extension
  ☆13Updated 9 months ago
• libeanim / contact-symplectic-integrator-network
  Contact-Aware Symplectic Integrator Network
  ☆14Updated 2 years ago
• PredictiveIntelligenceLab / JAX-BO
  ☆30Updated 2 years ago
• janosh / torch-mnf
  Multiplicative Normalizing Flows in PyTorch.
  ☆24Updated last month
• Zymrael / torchSODE
  PyTorch block-diagonal ODE CUDA solver, designed for gradient-based optimization
  ☆15Updated 5 years ago
• sscardapane / hydra-tutorial
  Companion code for a tutorial on using Hydra.
  ☆29Updated 3 years ago
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆62Updated 2 weeks ago
• facebookresearch / Addressing-the-Topological-Defects-of-Disentanglement
  Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"
  ☆22Updated 2 years ago
• lucidrains / tranception-pytorch
  Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction
  ☆31Updated 2 years ago
• tristandeleu / jax-comln
  Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)
  ☆26Updated 3 years ago
• DiffEqML / tutorials
  ☆10Updated 3 years ago
• facebookresearch / SparseBO
  code associated with paper "Sparse Bayesian Optimization"
  ☆26Updated last year
• ThomasMiconi / LearningToLearnCogTasks
  Code for the ICML 2023 paper "Learning to acquire novel cognitive tasks with evolution, plasticity and meta-meta-learning"
  ☆10Updated last year
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 3 years ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆49Updated 3 years ago