conda-forge / jaxlib-feedstock
A conda-smithy repository for jaxlib.
☆16Updated 2 weeks ago
Alternatives and similar repositories for jaxlib-feedstock:
Users that are interested in jaxlib-feedstock are comparing it to the libraries listed below
- Contact-Aware Symplectic Integrator Network☆12Updated 2 years ago
- Repository for DTU Special Course, focusing on Variational Inference using Normalizing Flows (VINF). Supervised by Michael Riis Andersen☆25Updated 4 years ago
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆9Updated 4 years ago
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆22Updated 2 years ago
- Multiplicative Normalizing Flows in PyTorch.☆24Updated 2 weeks ago
- Repository for the PopulAtion Parameter Averaging (PAPA) paper☆26Updated 11 months ago
- Automatically generate simple meta-learning tasks from a very large space☆15Updated last year
- Large-Scale Machine and Deep Learning in PyTorch.☆21Updated 3 years ago
- Applying Evolutionary Computing to Diffusion Models☆10Updated last week
- Quantification of Uncertainty with Adversarial Models☆28Updated last year
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- ☆34Updated last year
- Sharing Deep Learning Research Models with Lightning Part 1: Building A Super Resolution App☆15Updated 2 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆24Updated 4 years ago
- Scale-invariant Learning by Physics Inversion (NeurIPS 2022)☆10Updated 2 years ago
- Convex potential flows☆83Updated 3 years ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆27Updated 3 years ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated 2 years ago
- ☆30Updated 6 months ago
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆26Updated 11 months ago
- Perceiver (transformer variant) implemented in JAX and Flax☆11Updated 4 years ago
- Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight☆52Updated 3 years ago
- LaTeX source code for the slides☆23Updated 3 years ago
- ☆10Updated 2 years ago
- A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes☆25Updated 3 years ago
- High-performance tokenized language data-loader for Python C++ extension☆12Updated 8 months ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆87Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- Layered distributions using FLAX/JAX☆10Updated 4 years ago