JAX: Difference between revisions
imported>Samuela No edit summary |
imported>Samuela No edit summary |
||
| Line 9: | Line 9: | ||
let | let | ||
# Fork with jax/jaxlib. See https://github.com/NixOS/nixpkgs/pull/134894. | # Fork with jax/jaxlib. See https://github.com/NixOS/nixpkgs/pull/134894. | ||
pkgs = import (fetchTarball("https://github.com/samuela/nixpkgs/archive/ | pkgs = import (fetchTarball("https://github.com/samuela/nixpkgs/archive/ce58a33c2d5ff9f6cf295c6dc3fb439a467eec79.tar.gz")) {}; | ||
in pkgs.mkShell { | in pkgs.mkShell { | ||
buildInputs = with pkgs; [ | buildInputs = with pkgs; [ | ||
| Line 34: | Line 34: | ||
let | let | ||
# Fork with jax/jaxlib. See https://github.com/NixOS/nixpkgs/pull/134894. | # Fork with jax/jaxlib. See https://github.com/NixOS/nixpkgs/pull/134894. | ||
pkgs = import (fetchTarball("https://github.com/samuela/nixpkgs/archive/ | pkgs = import (fetchTarball("https://github.com/samuela/nixpkgs/archive/ce58a33c2d5ff9f6cf295c6dc3fb439a467eec79.tar.gz")) {}; | ||
in pkgs.mkShell { | in pkgs.mkShell { | ||
buildInputs = with pkgs; [ | buildInputs = with pkgs; [ | ||