The SciML Style Guide is a style guide for the Julia programming language. It is used by the SciML Open Source Scientific Machine Learning Organization. It covers proper styles to allow for easily high-quality, readable, robust, safety, and fast code that is easy to maintain for production and deployment.
It is open to discussion with the community. Please file an issue or open a PR to discuss changes to the style guide.
- SciML Style Guide for Julia
- Code Style Badge
- Overarching Dogmas of the SciML Style
- Consistency vs Adherence
- Community Contribution Guidelines
- Open source contributions are allowed to start small and grow over time
- Generic code is preferred unless code is known to be specific
- Internal types should match the types used by users when possible
- Trait definition and adherence to generic interface is preferred when possible
- Macros should be limited and only be used for syntactic sugar
- Errors should be caught as high as possible, and error messages should be contextualized for newcomers
- Subpackaging and interface packages is preferred over conditional modules via Requires.jl
- Functions should either attempt to be non-allocating and reuse caches, or treat inputs as immutable
- Out-Of-Place and Immutability is preferred when sufficient performant
- Tests should attempt to cover a wide gamut of input types
- When in doubt, a submodule should become a subpackage or separate package
- Globals should be avoided whenever possible
- Type-stable and Type-grounded code is preferred wherever possible
- Closures should be avoided whenever possible
- Numerical functionality should use the appropriate generic numerical interfaces
- Functions should capture one underlying principle
- Internal choices should be exposed as options whenever possible
- Prefer code reuse over rewrites whenever possible
- Prefer to not shadow functions
- Avoid unmaintained dependencies
- Avoid unsafe operations
- Avoid non public operations in Julia Base and packages
- Always default to constructs which initialize data
- Use extra precaution when running external processes
- Avoid eval whenever possible
- Avoid bounds check removal, and if done, add appropriate manual checks
- Avoid ccall unless necessary, and use safe ccall practices when required
- Validate all user inputs to avoid code injection
- Ensure secure random number generators are used when required
- Be aware of distributed computing encryption principles
- Always immediately flush secret data after handling
- Specific Rules
- High Level Rules
- General Naming Principles
- Comments
- Modules
- Functions
- Function Argument Precedence
- Tests and Continuous Integration
- Whitespace
- NamedTuples
- Numbers
- Ternary Operator
- For loops
- Function Type Annotations
- Struct Type Annotations
- Macros
- Types and Type Annotations
- Package version specifications
- Documentation
- Error Handling
- Arrays
- Line Endings
- VS-Code Settings
- JuliaFormatter
- References
Let contributors know your project is following the SciML Style Guide by adding the badge to your README.md.
[](https://github.com/SciML/SciMLStyle)According to PEP8:
A style guide is about consistency. Consistency with this style guide is important. Consistency within a project is more important. Consistency within one module or function is the most important.
But most importantly: know when to be inconsistent -- sometimes the style guide just doesn't apply. When in doubt, use your best judgment. Look at other examples and decide what looks best. And don't hesitate to ask!
Some code within the SciML organization is old, on life support, donated by researchers to be maintained. Consistency is the number one goal, so updating to match the style guide should happen on a repo-by-repo basis, i.e. do not update one file to match the style guide (leaving all other files behind).
For a comprehensive set of community contribution guidelines, refer to ColPrac. A relevant point to highlight PRs should do one thing. In the context of style, this means that PRs that update the style of a package's code should not be mixed with fundamental code contributions. This separation makes it easier to ensure that large style improvements are isolated from substantive (and potentially breaking) code changes.
If the standard for code contributions is that every PR needs to support every possible input type that anyone can
think of, the barrier would be too high for newcomers. Instead, the principle is to be as correct as possible to
begin with, and grow the generic support over time. All recommended functionality should be tested, and any known
generality issues should be documented in an issue (and with a @test_broken test when possible). However, a
function that is known to not be GPU-compatible is not grounds to block merging, rather it is encouraged for a
follow-up PR to improve the general type support!
For example, the code:
function f(A, B)
for i in 1:length(A)
A[i] = A[i] + B[i]
end
endwould not be preferred for two reasons. One is that it assumes A uses one-based indexing, which would fail in cases
like OffsetArrays and FFTViews.
Another issue is that it requires indexing, while not all array types support indexing (for example,
CuArrays). A more generic and compatible implementation of this function would be
to use broadcast, for example:
function f(A, B)
@. A = A + B
endwhich would allow support for a wider variety of array types.
If f(A) takes the input of some collections and computes an output from those collections, then it should be
expected that if the user gives A as an Array, the computation should be done via Arrays. If A was a
CuArray, then it should be expected that the computation should be internally done using a CuArray (or appropriately
error if not supported). For these reasons, constructing arrays via generic methods, like similar(A), is preferred when
writing f instead of using non-generic constructors like Array(undef,size(A)) unless the function is documented as
being non-generic.
Julia provides many different interfaces, for example:
Those interfaces should be followed when possible. For example, when defining broadcast overloads,
one should implement a BroadcastStyle as suggested by the documentation instead of simply attempting
to bypass the broadcast system via copyto! overloads.
When interface functions are missing, these should be added to Base Julia or an interface package,
like ArrayInterface.jl. Such traits should be
declared and used when appropriate. For example, if a line of code requires mutation, the trait
ArrayInterface.ismutable(A) should be checked before attempting to mutate, and informative error
messages should be written to capture the immutable case (or, an alternative code that does not
mutate should be given).
One example of this principle is demonstrated in the generation of Jacobian matrices. In many scientific
applications, one may wish to generate a Jacobian cache from the user's input u0. A naive way to generate
this Jacobian is J = similar(u0,length(u0),length(u0)). However, this will generate a Jacobian J such
that J isa Matrix.
Macros define new syntax, and for this reason, they tend to be less composable than other coding styles and require prior familiarity to be easily understood. One principle to keep in mind is, “can the person reading the code easily picture what code is being generated?”. For example, a user of Soss.jl may not know what code is being generated by:
@model (x, α) begin
σ ~ Exponential()
β ~ Normal()
y ~ For(x) do xj
Normal(α + β * xj, σ)
end
return y
endand thus using such a macro as the interface is not preferred when possible. However, a macro like
@muladd is trivial to picture on a code (it recursively
transforms a*b + c to muladd(a,b,c) for more
accuracy and efficiency), so using
such a macro, for example:
julia> @macroexpand(@muladd k3 = f(t + c3 * dt, @. uprev + dt * (a031 * k1 + a032 * k2)))
:(k3 = f((muladd)(c3, dt, t), (muladd).(dt, (muladd).(a032, k2, (*).(a031, k1)), uprev)))is recommended. Some macros in this category are:
Some performance macros, like @simd, @threads, or
@turbo from LoopVectorization.jl,
make an exception in that their generated code may be foreign to many users. However, they still are
classified as appropriate uses as they are syntactic sugar since they do (or should) not change the behavior
of the program in measurable ways other than performance.
Errors should be caught as high as possible, and error messages should be contextualized for newcomers
Whenever possible, defensive programming should be used to check for potential errors before they are encountered
deeper within a package. For example, if one knows that f(u0,p) will error unless u0 is the size of p, this
should be caught at the start of the function to throw a domain specific error, for example “parameters and initial
condition should be the same size”.
This contextualization should result in error messages that use terminology related to the user facing API (vs. referencing internal implementation details). Ideally, such error messages should not only describe the issue in language that will be familiar to the user but also include suggestions, where possible, of how to correct the issue.
Requires.jl should be avoided at all costs. If an interface package exists, such as ChainRulesCore.jl for defining automatic differentiation rules without requiring a dependency on the whole ChainRules.jl system, or RecipesBase.jl which allows for defining Plots.jl plot recipes without a dependency on Plots.jl, a direct dependency on these interface packages is preferred.
Otherwise, instead of resorting to a conditional dependency using Requires.jl, it is preferred to create subpackages, i.e. smaller independent packages kept within the same Github repository with independent versioning and package management. An example of this is seen in Optimization.jl which has subpackages like OptimizationBBO.jl for BlackBoxOptim.jl support.
Some important interface packages to know about are:
Mutating codes and non-mutating codes fall into different worlds. When a code is fully immutable, the compiler can better reason about dependencies, optimize the code, and check for correctness. However, many times a code that makes the fullest use of mutation can outperform even what the best compilers of today can generate. That said, the worst of all worlds is when code mixes mutation with non-mutating code. Not only is this a mishmash of coding styles, but it also has the potential non-locality and compiler proof issues of mutating code while not fully benefiting from the mutation.
Mutation is used to get more performance by decreasing the number of heap allocations. However,
if it's not helpful for heap allocations in a given spot, do not use mutation. Mutation is scary
and should be avoided unless it gives an immediate benefit. For example, if
matrices are sufficiently large, then A*B is as fast as mul!(C,A,B), and thus writing
A*B is preferred (unless the rest of the function is being careful about being fully non-allocating,
in which case this should be mul! for consistency).
Similarly, when defining types, using struct is preferred to mutable struct unless mutating
the struct is a common occurrence. Even if mutating the struct is a common occurrence, see whether
using Setfield.jl is sufficient. The compiler will optimize
the construction of immutable structs, and thus this can be more efficient if it's not too much of a
code hassle.
Code coverage numbers are meaningless if one does not consider the input types. For example, one can
hit all the code with Array, but that does not test whether CuArray is compatible! Thus it's
always good to think of coverage not in terms of lines of code but in terms of type coverage. A good
list of number types to think about are:
Float64Float32ComplexDualBigFloat
Array types to think about testing are:
ArrayOffsetArrayCuArray
Keep packages focused on one core idea. If there's something separate enough to be a submodule, could it instead be a separate, well-tested and documented package to be used by other packages? Most likely yes.
Global variables should be avoided whenever possible. When required, global variables should be
constants and have an all uppercase name separated with underscores (e.g. MY_CONSTANT). They should be
defined at the top of the file, immediately after imports and exports but before an __init__ function.
If you truly want mutable global style behavior you may want to look into mutable containers.
Type-stable and type-grounded code helps the compiler create not only more optimized code, but also faster to compile code. Always keep containers well-typed, functions specializing on the appropriate arguments, and types concrete.
Closures can cause accidental type instabilities that are difficult to track down and debug; in the
long run, it saves time to always program defensively and avoid writing closures in the first place,
even when a particular closure would not have been problematic. A similar argument applies to reading
code with closures; if someone is looking for type instabilities, this is faster to do when code does
not contain closures.
Furthermore, if you want to update variables in an outer scope, do so explicitly with Refs or self
defined structs.
For example,
map(Base.Fix2(getindex, i), vector_of_vectors)is preferred over
map(v -> v[i], vector_of_vectors)or
[v[i] for v in vector_of_vectors]While you can use A\b to do a linear solve inside a package, that does not mean that you should.
This interface is only sufficient for performing factorizations, and so that limits the scaling
choices, the types of A that can be supported, etc. Instead, linear solves within packages should
use LinearSolve.jl. Similarly, nonlinear solves should use NonlinearSolve.jl. Optimization should use
Optimization.jl. Etc. This allows the full generic choice to be given to the user without depending
on every solver package (effectively recreating the generic interfaces within each package).
Functions mean one thing. Every dispatch of + should be “the meaning of addition on these types”.
While in theory you could add dispatches to + that mean something different, that will fail in
generic code for which + means addition. Thus, for generic code to work, code needs to adhere to
one meaning for each function. Every dispatch should be an instantiation of that meaning.
Whenever possible, numerical values and choices within scripts should be exposed as options to the user. This promotes code reusability beyond the few cases the author may have expected.
If a package has a function you need, use the package. Add a dependency if you need to. If the function is missing a feature, prefer to add that feature to said package and then add it as a dependency. If the dependency is potentially troublesome, for example because it has a high load time, prefer to spend time helping said package fix these issues and add the dependency. Only when it does not seem possible to make the package “good enough” should using the package be abandoned. If it is abandoned, consider building a new package for this functionality as you need it, and then make it a dependency.
Two functions can have the same name in Julia by having different namespaces. For example,
X.f and Y.f can be two different functions, with different dispatches, but the same name.
This should be avoided whenever possible. Instead of creating MyPackage.sort, consider
adding dispatches to Base.sort for your types if these new dispatches match the underlying
principle of the function. If it doesn't, prefer to use a different name. While using MyPackage.sort
is not conflicting, it is going to be confusing for most people unfamiliar with your code,
so MyPackage.special_sort would be more helpful to newcomers reading the code.
Packages should only be depended on if they have maintainers who are responsive. Good code requires good communities. If maintainers do not respond to breakage within 2 weeks with multiple notices, then all dependencies from that organization should be considered for removal. Note that some issues may take a long time to fix, so it may take more time than 2 weeks to fix, it's simply that the communication should be open, consistent, and timely.
Like other high-level languages that provide strong safety guarantees by default, Julia nevertheless has a small
set of operations that bypass normal checks. These operations are clearly marked with the prefix unsafe_. By using an
“unsafe” operation, the programmer asserts that they know the operation is valid even though the language cannot
automatically ensure it. For high reliability these constructs should be avoided or carefully inspected during code
review. They are:
unsafe_loadunsafe_store!unsafe_readunsafe_writeunsafe_stringunsafe_wrapunsafe_convertunsafe_copyto!unsafe_pointer_to_objrefccall@ccall@inbounds
The Julia standard library and packages developed in the Julia programming language have an intended public API indicated by
marking symbols with the export keyword or in v1.11+ with the new public keyword.
However, it is possible to use non public names via explicit qualification, e.g. Base.foobar. This practice is not necessarily unsafe,
but should be avoided since non public operations may have unexpected invariants and behaviors, and are subject to changes in future
releases of the language.
Note that qualified names are commonly used in method definitions to clarify that a function is being extended, e.g. function Base.getindex(...) … end.
Such uses do not fall under this concern.
For certain newly-allocated data structures, such as numeric arrays, the Julia compiler and runtime do not check whether data is accessed before it has been
initialized. Therefore such data structures can “leak” information from one part of a program to another. Uninitialized structures should be avoided in favor
of functions like zeros and fill that create data with well-defined contents. If code does allocate uninitialized memory, it should ensure that this memory
is fully initialized before being returned from the function in which it is allocated.
Constructs which create uninitialized memory should only be used if there is a demonstrated performance impact and it should ensure that all memory is initialized in the same function in which the array is intended to be used.
Example:
function make_array(n::Int)
A = Vector{Int}(undef, n)
# function body
return A
endThis function allocates an integer array with undefined initial contents (note the language forces you to request this explicitly). A code reviewer should ensure that the function body assigns every element of the array. One can similarly create structs with undefined fields, and if used this way, one should ensure all fields are initialized:
struct Foo
x::Int
Foo() = new()
end
julia> Foo().x
139736495677280The Julia standard library contains a run function and other facilities for running external processes. Any program that does this
is only as safe as the external process it runs. If this cannot be avoided, then best practices for using these features are:
- Only run fixed, known executables, and do not derive the path of the executable to run from user input or other outside sources.
- Make sure the executables used have also passed required audit procedures.
- Make sure to handle process failure (non-zero exit code).
- If possible, run external processes in a sandbox or “jail” environment with access only to what they need in terms of files, file handles and ports.
When run in a sandbox or jail, external processes can actually improve security since the subprocess is isolated from the rest of the system by the kernel.